| # data | |
| augmentations: 'vit_heavy' | |
| image_size: [256, 256] # [height, width] | |
| dataset: 'SnakeCLEF2023' | |
| # model | |
| architecture: 'swinv2_tiny_window16_256.ms_in1k' | |
| # training | |
| loss: 'SeeSawLoss' | |
| optimizer: 'SGD' | |
| scheduler: 'plateau' | |
| epochs: 100 | |
| learning_rate: 0.01 | |
| batch_size: 32 | |
| accumulation_steps: 4 | |
| # other | |
| random_seed: 777 | |
| workers: 1 | |
| multigpu: False | |
| tags: ["Fine-tuning"] # W&B Run tags | |
| root_path: "./" |