jpterry commited on
Commit
bf213f6
·
1 Parent(s): 4cddfc0
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -83,7 +83,7 @@ def get_activations(model, image: list, model_name: str,
83
 
84
 
85
  layer_outputs = {}
86
- for i in range(len(model.model.features)):
87
  image = model.features[i](image)
88
  layer_outputs[i] = image
89
  print(i, layer_outputs[i].shape)
 
83
 
84
 
85
  layer_outputs = {}
86
+ for i in range(len(model.features)):
87
  image = model.features[i](image)
88
  layer_outputs[i] = image
89
  print(i, layer_outputs[i].shape)