34587tresrtyuiodsadfiop[
Browse files
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
|