Compositional vision transformers training

2023-09-25 17:45:19 +02:00 · 2023-09-25 17:45:19 +02:00 · 35d50c5e2b
parent 8ad6e54e68
commit 35d50c5e2b
1 changed files with 2 additions and 0 deletions
--- a/vision_transformer.py
+++ b/vision_transformer.py
@ -157,6 +157,8 @@ class CompVisionTransformer(nn.Module):
            ]
        else:
            output = x[:, 0, :]
+        print('!!!')
+        print(output)
        output = self.pre_logits(output)
        return output