From 25241cf44dc93bdef2e386c45da10a003d701f86 Mon Sep 17 00:00:00 2001 From: p82b7rtam <1761133400@qq.com> Date: Mon, 29 Dec 2025 16:38:42 +0800 Subject: [PATCH] ADD file via upload --- default.yaml | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 default.yaml diff --git a/default.yaml b/default.yaml new file mode 100644 index 0000000..8c24c2e --- /dev/null +++ b/default.yaml @@ -0,0 +1,36 @@ +# configs/stable_config.yaml +# 稳定训练配置 + +# 模型配置 +model: + model_type: "vgg16" + image_size: 192 + content_layers: ["19"] # conv4_2 + style_layers: ["0", "5", "10", "17", "24"] # 全部5个层 + content_weight: 1e4 # 增加内容权重 + style_weight: 5e4 # 降低风格权重 + learning_rate: 0.005 # 低学习率稳定训练 + +# 路径配置 +paths: + content_image: "data/raw/content.jpg" + style_image: "data/raw/tem.jpg" + output_dir: "data/processed" + +# 训练配置 +training: + num_steps: 150 # 减少步数 + log_interval: 15 + tv_weight: 1e-5 + +# 多尺度配置(先禁用,稳定后再启用) +multi_scale: + enabled: false + scales: [192] + scale_weights: [1.0] + +# 性能配置 +performance: + use_gpu: false + batch_size: 1 + precision: "float32" \ No newline at end of file