jpterry commited on
Commit
cce295c
·
1 Parent(s): 55c91f7

print statements

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -129,6 +129,8 @@ def get_activations(model, image: list, model_name: str,
129
  output_1 = layer_outputs[activation_indices[model_name][0]].detach().cpu().numpy()
130
  output_2 = layer_outputs[activation_indices[model_name][1]].detach().cpu().numpy()
131
 
 
 
132
  # get activations
133
  # output_1 = outputs[1]
134
  # output_2 = outputs[2]
 
129
  output_1 = layer_outputs[activation_indices[model_name][0]].detach().cpu().numpy()
130
  output_2 = layer_outputs[activation_indices[model_name][1]].detach().cpu().numpy()
131
 
132
+ print(image.shape, output.shape, output_1.shape, output_2.shape)
133
+
134
  # get activations
135
  # output_1 = outputs[1]
136
  # output_2 = outputs[2]