tensorgirl
/

TFaugvit

Image Classification

Model card Files Files and versions

tensorgirl commited on Sep 21, 2023

Commit

37c96c3

·

1 Parent(s): ac9d0d5

Update augvit_config.py

Files changed (1) hide show

augvit_config.py +1 -1

augvit_config.py CHANGED Viewed

@@ -11,7 +11,7 @@ class AugViTConfig(PretrainedConfig):
         patch_size: int = 32,
         num_classes: int = 1000,
         dim: int = 128,
-        depth: int = 6,
         heads: int = 16,
         mlp_dim: int = 256,
         dropout: int = 0.1,

         patch_size: int = 32,
         num_classes: int = 1000,
         dim: int = 128,
+        depth: int = 2,
         heads: int = 16,
         mlp_dim: int = 256,
         dropout: int = 0.1,