Update augvit_config.py

augvit_config.py

@@ -7,9 +7,9 @@ class AugViTConfig(PretrainedConfig):
 
     def __init__(
         self,
-        image_size: int = 32,
-        patch_size: int = 4,
-        num_classes: int = 10,
+        image_size: int = 224,
+        patch_size: int = 32,
+        num_classes: int = 1000,
         dim: int = 128,
         depth: int = 6,
         heads: int = 16,