【创新实训】风格迁移功能探索与实现（六）完成代码并训练 保存训练结果 测试效果

全部完成时代码结构如下：

 使用MS COCO数据集：

放到train2014下。

该数据集解压后18G左右，共328000张图片，每张style-image训练的话，在服务器GPU环境下需要4-5个小时左右，迭代次数50000多次，最终Loss收敛在六七十。

测试使用模型：

可以看到，使用cubist风格转换后还不错

最终训练结果：