| seed_everything: false | |
| trainer: | |
| accelerator: gpu | |
| devices: 1 | |
| precision: 16-mixed | |
| max_epochs: 75 | |
| logger: | |
| class_path: lightning.pytorch.loggers.TensorBoardLogger | |
| init_args: | |
| save_dir: logs/resnet18 | |
| name: packed | |
| default_hp_metric: false | |
| model: | |
| num_classes: 10 | |
| in_channels: 3 | |
| loss: CrossEntropyLoss | |
| version: packed | |
| arch: 18 | |
| style: cifar | |
| num_estimators: 4 | |
| alpha: 2 | |
| gamma: 1 | |
| pretrained: true | |
| data: | |
| root: ./data | |
| batch_size: 128 | |
| optimizer: | |
| lr: 0.05 | |
| momentum: 0.9 | |
| weight_decay: 5e-4 | |
| nesterov: true | |
| lr_scheduler: | |
| milestones: | |
| - 25 | |
| - 50 | |
| gamma: 0.1 | |