Spaces:
Runtime error
Runtime error
File size: 313 Bytes
cf754af 4ff4028 cf754af de7d21e cf754af de7d21e cf754af de7d21e cf754af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# model class
_target_: src.models.catdog_model.ViTTinyClassifier
# model params
img_size: ${data.image_size}
patch_size: 16
num_classes: 2
embed_dim: 128
depth: 6
num_heads: 4
mlp_ratio: 4
pre_norm: False
# optimizer params
lr: 1e-3
weight_decay: 1e-5
# scheduler params
factor: 0.1
patience: 10
min_lr: 1e-6 |