| #网页设置 | |
| wandb_setup: | |
| project: "cifar10_transfer_learning" | |
| experiment: "ResNet50_ImageNet_224px" | |
| tags: ["cifar10","resnet50"] | |
| seed: 42 | |
| job_type: "train" | |
| #数据加载参数 | |
| data: | |
| data_path: "./data" | |
| batch_size: 64 | |
| num_workers: 4 | |
| image_size: [224,224] | |
| in_channels: 3 | |
| #模型结构参数 | |
| model: | |
| type: "TransferResNet50" | |
| dropout_rate: 0.0 | |
| num_classes: 10 | |
| #训练超参数 | |
| train: | |
| epochs: 30 | |
| save_dir: "./models" | |
| #优化器与调度器 | |
| optimizer: | |
| name: "adamw" | |
| lr: 0.001 | |
| backbone_lr: 0.00005 | |
| weight_decay: 1e-3 | |
| scheduler: | |
| use_scheduler: True | |
| type: "CosineAnnealingLR" | |
| T_max: 30 | |
| eta_min: 1e-6 | |