We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 87488e2 commit 1e8bd69Copy full SHA for 1e8bd69
bayesian_torch/models/bnn_to_qbnn.py
@@ -165,8 +165,8 @@ def qbnn_batchnorm2d_layer(d):
165
# qbnn_layer.bias = Parameter(get_quantized_tensor(d.bias), requires_grad=False)
166
# qbnn_layer.running_mean = Parameter(get_quantized_tensor(d.running_mean), requires_grad=False)
167
# 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)
+ # qbnn_layer.scale = Parameter(torch.tensor([0.1]), requires_grad=False)
+ # qbnn_layer.zero_point = Parameter(torch.tensor([128]), requires_grad=False)
170
return qbnn_layer
171
172
0 commit comments