预训练模型下载

[miraiwk]

这里提供预训练模型的下载

模型：UGATIT Light
数据集：selfie2anime
Batch Size: 1
总迭代数: 100万

预训练模型下载:
链接: https://pan.baidu.com/s/1ukOd1xznJn8H7IkMUxcbug 密码: t7ft

压缩包最后的数字表示迭代次数, 数字越大表示迭代次数越多, 下载一个压缩包就好. 模型权重文件持续更新中.

把文件解压后， 复制里面所有的*.bdparams文件到文件夹results/<数据集名称>/model/下就可以了。

继续训练模型的脚本：

FLAGS_cudnn_exhaustive_search=True python main.py --light True --save_freq 500 --dataset data48778 --print_freq 200 --resume True

记得修改dataset参数中的数据集名称。

[TomatoBoy90]

太棒了，我还想着自己用paddle写，没想到你都写好了，还有预训练模型