tensorgirl
/

TFaugvit

Image Classification

Inference Endpoints

Model card Files Files and versions Community

tensorgirl commited on Sep 21, 2023

Commit

81dc951

·

1 Parent(s): a221e9d

Update augvit_model.py

Files changed (1) hide show

augvit_model.py +1 -1

augvit_model.py CHANGED Viewed

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

         # )
         # x = tf.concat([cls_tokens, x], axis=1)
         # print(x.shape,cls_tokens.shape )
+        x += self.pos_embedding[:, :(n )]
         print(x.shape)
         x = self.dropout(x, training=training)
         print(x.shape)