morinop commited on
Commit
02d0cf1
1 Parent(s): 719452d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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),numpy().tolist())
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