Update augvit_model.py

augvit_model.py

@@ -141,8 +141,8 @@ class AUGViT(Model):
         )
         x = tf.concat([cls_tokens, x], axis=1)
         print(x.shape,cls_tokens.shape )
-        pos= self.pos_embedding[:, :(n+1 )]
-        x += pos
+        # pos= self.pos_embedding[:, :(n+1 )]
+        # x += pos
         
         print(x.shape,pos.shape,self.pos_embedding.shape)
         x = self.dropout(x, training=training)