30 lines
1.2 KiB
YAML
30 lines
1.2 KiB
YAML
# 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 # 是否启用分布式训练
|
||
dataset: "./dataset_electornic.txt" # 数据集名称
|
||
|
||
|
||
# 模型配置
|
||
models:
|
||
backbone: 'resnet101'
|
||
channel_ratio: 1.0
|
||
model_path: "../checkpoints/resnet101_electornic_20250807/best.pth"
|
||
onnx_model: "../checkpoints/resnet101_electornic_20250807/best.onnx"
|
||
rknn_model: "../checkpoints/resnet101_electornic_20250807/resnet101_electornic.rknn"
|
||
rknn_batch_size: 1
|
||
|
||
# 日志与监控
|
||
logging:
|
||
logging_dir: "./logs" # 日志保存目录
|
||
tensorboard: true # 是否启用TensorBoard
|
||
checkpoint_interval: 30 # 检查点保存间隔（epoch）
|