修改Dataloader提升训练效率
This commit is contained in:
@ -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)
|
||||
|
||||
|
Reference in New Issue
Block a user