| { | |
| "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" | |
| } | |