修改Dataloader提升训练效率

This commit is contained in:
lee
2025-08-07 10:52:42 +08:00
parent 3392d76e38
commit ebba07d1ca
3 changed files with 98 additions and 48 deletions

View File

@ -15,7 +15,7 @@ base:
# 模型配置
models:
backbone: 'resnet18'
backbone: 'resnet50'
channel_ratio: 1.0
# 训练参数
@ -31,7 +31,7 @@ training:
weight_decay: 0.0005 # 权重衰减
scheduler: "step" # 学习率调度器可选cosine/cosine_warm/step/None
num_workers: 32 # 数据加载线程数
checkpoints: "./checkpoints/resnet18_electornic_20250806/" # 模型保存目录
checkpoints: "./checkpoints/resnet50_electornic_20250807/" # 模型保存目录
restore: false
restore_model: "./checkpoints/resnet18_20250717_scale=0.75_nosub/best.pth" # 模型恢复路径
cosine_t_0: 10 # 初始周期长度
@ -62,7 +62,7 @@ transform:
# 日志与监控
logging:
logging_dir: "./logs/resnet18_scale=0.75_nosub_log" # 日志保存目录
logging_dir: "./logs/resnet50_electornic_log" # 日志保存目录
tensorboard: true # 是否启用TensorBoard
checkpoint_interval: 30 # 检查点保存间隔epoch