Spaces:
Build error
Build error
print statements
Browse files
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]
|