# 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_3588.rknn" rknn_batch_size: 1 # 日志与监控 logging: logging_dir: "./logs" # 日志保存目录 tensorboard: true # 是否启用TensorBoard checkpoint_interval: 30 # 检查点保存间隔(epoch)