| DATASET: | |
| data_path: "./data/head/experiment_5/prop_0_0/" | |
| train_path: "./data/head/experiment_5/prop_0_0/train_80" | |
| val_path: "./data/head/experiment_5/prop_0_0/val_10" | |
| test_path: "./data/head/experiment_5/prop_0_0/test_10" | |
| num_class: 4 | |
| img_size: 299 | |
| MODEL: | |
| name: "xception" | |
| ckpt_path: "./ckpt/head/experiment_5/" | |
| model_path: "./models/head/experiment_5/" | |
| preds_path: "./predictions/head/experiment_5/" | |
| plots_path: "./plots/head/experiment_5/" | |
| TRAIN: | |
| batch_size: 32 | |
| num_epoch: 200 | |
| augment: True | |
| optimizer: "Adam" | |
| lr: 0.001 | |
| lr_tune: 0.0001 | |
| es: 20 | |