File size: 448 Bytes
ba0c2f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
{
"data_path": "saved/data/fer2013",
"image_size": 224,
"in_channels": 3,
"num_classes": 7,
"arch": "alexnet",
"lr": 0.0001,
"weighted_loss": 0,
"momentum": 0.9,
"weight_decay": 0.001,
"distributed": 0,
"batch_size": 48,
"num_workers": 8,
"device": "cuda:0",
"max_epoch_num": 50,
"max_plateau_count": 8,
"plateau_patience": 2,
"steplr": 50,
"log_dir": "saved/logs",
"checkpoint_dir": "saved/checkpoints",
"model_name": "_n"
}
|