You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
837 B
36 lines
837 B
# 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" |