Mojmir commited on
Commit
0e6c4c2
1 Parent(s): e1514f1

34587tresrtyuiodsadfiop[

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def load_secure_model(model):
27
  secure_model = compile_torch_model(
28
  model.to("cpu"),
29
  n_bits={"model_inputs": 4, "op_inputs": 3, "op_weights": 3, "model_outputs": 5},
30
- rounding_threshold_bits={"n_bits": 7, "method": "APPROXIMATE"},
31
  torch_inputset=torch.rand(10, 3, 224, 224)
32
  )
33
  return secure_model
 
27
  secure_model = compile_torch_model(
28
  model.to("cpu"),
29
  n_bits={"model_inputs": 4, "op_inputs": 3, "op_weights": 3, "model_outputs": 5},
30
+ # rounding_threshold_bits={"n_bits": 7, "method": "APPROXIMATE"},
31
  torch_inputset=torch.rand(10, 3, 224, 224)
32
  )
33
  return secure_model