增加学习率调度方式

2025-06-13 10:45:53 +08:00
parent 37ecef40f7
commit 1803f319a5
13 changed files with 319 additions and 294 deletions
--- a/configs/transform.yml
+++ b/configs/transform.yml
@ -0,0 +1,27 @@
+# configs/transform.yml
+# pth转换onnx配置文件
+
+# 基础配置
+base:
+  experiment_name: "model_comparison"  # 实验名称（用于结果保存目录）
+  seed: 42                            # 随机种子（保证可复现性）
+  device: "cuda"                      # 训练设备（cuda/cpu）
+  log_level: "info"                   # 日志级别（debug/info/warning/error）
+  embedding_size: 256                 # 特征维度
+  pin_memory: true                    # 是否启用pin_memory
+  distributed: true                  # 是否启用分布式训练
+
+
+# 模型配置
+models:
+  backbone: 'resnet50'
+  channel_ratio: 1.0
+  model_path: "../checkpoints/resnet50_0519/best.pth"
+  onnx_model: "../checkpoints/resnet50_0519/best.onnx"
+  rknn_model: "../checkpoints/resnet50_0519/best.rknn"
+
+# 日志与监控
+logging:
+  logging_dir: "./logs"                  # 日志保存目录
+  tensorboard: true                   # 是否启用TensorBoard
+  checkpoint_interval: 30             # 检查点保存间隔（epoch）