Skip to content

Commit 1e8bd69

Browse files
committed
fix accuracy drop
1 parent 87488e2 commit 1e8bd69

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bayesian_torch/models/bnn_to_qbnn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,8 @@ def qbnn_batchnorm2d_layer(d):
165165
# qbnn_layer.bias = Parameter(get_quantized_tensor(d.bias), requires_grad=False)
166166
# qbnn_layer.running_mean = Parameter(get_quantized_tensor(d.running_mean), requires_grad=False)
167167
# qbnn_layer.running_var = Parameter(get_quantized_tensor(d.running_var), requires_grad=False)
168-
qbnn_layer.scale = Parameter(torch.tensor([0.1]), requires_grad=False)
169-
qbnn_layer.zero_point = Parameter(torch.tensor([128]), requires_grad=False)
168+
# qbnn_layer.scale = Parameter(torch.tensor([0.1]), requires_grad=False)
169+
# qbnn_layer.zero_point = Parameter(torch.tensor([128]), requires_grad=False)
170170
return qbnn_layer
171171

172172

0 commit comments

Comments
 (0)