parent
258c7111fd
commit
25241cf44d
@ -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"
|
||||
Loading…
Reference in new issue