vit_model/config.json

{
  "architecture": "swinv2_base_window12_192",
  "pretrained": true,
  "num_classes": 15,
  "input_size": 192,
  "patch_size": 4,
  "window_size": 12,
  "embed_dim": 128,
  "depths": [2, 2, 18, 2],
  "num_heads": [4, 8, 16, 32],
  "mlp_ratio": 4.0,
  "qkv_bias": true,
  "drop_rate": 0.0,
  "attn_drop_rate": 0.0,
  "drop_path_rate": 0.2,
  "batch_size": 34,
  "learning_rate": 2e-5,
  "weight_decay": 5e-4,
  "epochs": 18,
  "use_amp": true
}