tensorgirl
/

TFaugvit

Image Classification

Model card Files Files and versions

tensorgirl commited on Sep 21, 2023

Commit

4e2f8db

·

1 Parent(s): d4aa7f8

Update augvit_model.py

Files changed (1) hide show

augvit_model.py +1 -1

augvit_model.py CHANGED Viewed

@@ -144,7 +144,7 @@ class AUGViT(Model):
         # pos= self.pos_embedding[:, :(n+1 )]
         # x += pos
-        print(x.shape,pos.shape,self.pos_embedding.shape)
         x = self.dropout(x, training=training)
         print(x.shape)

         # pos= self.pos_embedding[:, :(n+1 )]
         # x += pos
+        # print(x.shape,pos.shape,self.pos_embedding.shape)
         x = self.dropout(x, training=training)
         print(x.shape)