×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

学习经验 sarom2022-06-30 12:48:11 回复 2 查看 经验交流
学习经验
分享到:

非实时图像画风迁移使用的是经典的VGG模型,其网络结构如下:

实验中的 _1实现了全连接层、ReLU层、Softmax层, _2实现了卷积层和池化层以及加速算法, _3实现了内容损失和风格损失,exp_3_3_style_transfer则是搭建风格迁移网络并实现了AdamOptimizer优化器。补全代码后执行main_exp_3_3.py即可运行实验,每一轮的耗时大约在70分钟左右,全部炼完需要近5天。

在训练的过程中,如果断开了SSH连接,会在一段时间后结束进程,因此有两种方案:一是不关闭远程连接,而是不挂断地运行命令

方案二采用了nohup命令,执行nohup python main_exp_3_3.py后会生成nohup.out文件,打开发现程序执行正常,即可在终端输入exit命令退出连接,程序会在后台正常执行。如果要结束运行,可以查找查询的PID后使用kill命令,也可以一开始就写好脚本,那样执行和结束更为方便。

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭