Maintenance.

This commit is contained in:
Marcel Keller
2023-08-14 18:29:08 +10:00
parent 7bc156e581
commit 2813c0ef0f
140 changed files with 1598 additions and 388 deletions

View File

@@ -38,7 +38,7 @@ except:
try:
batch_size = int(program.args[2])
except:
batch_size = N
batch_size = min(N, 128)
if 'savemem' in program.args:
N = batch_size

View File

@@ -38,7 +38,7 @@ except:
try:
batch_size = int(program.args[2])
except:
batch_size = N
batch_size = min(N, 128)
if 'savemem' in program.args:
N = batch_size

View File

@@ -38,7 +38,7 @@ except:
try:
batch_size = int(program.args[2])
except:
batch_size = N
batch_size = min(N, 128)
assert batch_size <= N
ml.Layer.back_batch_size = batch_size

View File

@@ -26,12 +26,14 @@ exec(subprocess.check_output(['Scripts/process-tf.py', program.args[1]]))
opt = ml.Optimizer()
opt.set_layers_with_inputs(layers)
layers[0].X.input_from(0)
layers[0].X.input_from(0, binary=True)
for layer in layers:
layer.input_from(0, raw='raw' in program.args)
layer.input_from(0, binary=True)
sint(0).reveal().store_in_mem(0)
opt.time_layers = 'time_layers' in program.args
start_timer(1)
opt.forward(1, keep_intermediate=False)
stop_timer(1)