Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def print_bn():
|
|
22 |
bn_data.extend(m.running_var.data.numpy().tolist())
|
23 |
bn_data.append(m.momentum)
|
24 |
print(len(bn_data))
|
25 |
-
bn_data.extend(model.resnet.embedder.embedder.convolution.weight.data.reshape(-1)
|
26 |
# print(model.resnet.embedder.embedder.convolution.weight.data.numpy().tolist())
|
27 |
return bn_data
|
28 |
|
|
|
22 |
bn_data.extend(m.running_var.data.numpy().tolist())
|
23 |
bn_data.append(m.momentum)
|
24 |
print(len(bn_data))
|
25 |
+
bn_data.extend(model.resnet.embedder.embedder.convolution.weight.data.reshape(-1).numpy().tolist())
|
26 |
# print(model.resnet.embedder.embedder.convolution.weight.data.numpy().tolist())
|
27 |
return bn_data
|
28 |
|