File size: 423 Bytes
a0f970c |
1 |
{"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"} |