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
|