tensorgirl
/

TFaugvit

Image Classification

Model card Files Files and versions

tensorgirl commited on Sep 20, 2023

Commit

ede0b4d

·

1 Parent(s): 65a341c

Update augvit_model.py

Files changed (1) hide show

augvit_model.py +1 -1

augvit_model.py CHANGED Viewed

@@ -182,5 +182,5 @@ class AugViTForImageClassification(TFPreTrainedModel):
         inp = pixel_values['pixel_values']
         if inp.shape[-1]!=3:
             inp = tf.transpose(inp,[0,2,3,1])
-        logits = self.model()
         return logits

         inp = pixel_values['pixel_values']
         if inp.shape[-1]!=3:
             inp = tf.transpose(inp,[0,2,3,1])
+        logits = self.model(inp)
         return logits