config.yaml

#网页设置
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