| data_params: | |
| data_path: "dataset" | |
| image_size: 224 | |
| model_params: | |
| name: "efficientnet_b0" | |
| pretrained: True | |
| train_params: | |
| epochs: 25 | |
| batch_size: 64 | |
| optimizer: "AdamW" | |
| learning_rate: 0.001 | |
| unfreeze_epoch: 5 | |
| ddp_params: | |
| master_port: '12355' | |
| output_params: | |
| save_dir: "runs/staging_classifier" | |
| checkpoint_name: "best_model.pth" | |