mirror of
https://github.com/data61/MP-SPDZ.git
synced 2026-01-10 14:08:09 -05:00
Maintenance.
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user