|
|
@ -41,10 +41,9 @@ if __name__ == "__main__": |
|
|
|
# limit gram alloc |
|
|
|
# limit gram alloc |
|
|
|
gpus = tf.config.experimental.list_physical_devices('GPU') |
|
|
|
gpus = tf.config.experimental.list_physical_devices('GPU') |
|
|
|
if len(gpus) > 0: |
|
|
|
if len(gpus) > 0: |
|
|
|
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)]) |
|
|
|
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)]) |
|
|
|
|
|
|
|
|
|
|
|
m = load_model(sys.argv[1]) |
|
|
|
m = load_model(sys.argv[1]) |
|
|
|
print(m, file=sys.stderr) |
|
|
|
print(m, file=sys.stderr) |
|
|
|
|
|
|
|
|
|
|
|
run_loop(m) |
|
|
|
run_loop(m) |
|
|
|
|
|
|
|
|
|
|
|