# 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"