search

hzy46 / fast-neural-style-tensorflow

A tensorflow implementation for fast neural style!
932 stars 361 forks source link

restoring parameters from pretrained/vgg_16.ckpt #31

Open JunweiSUN opened 7 years ago

JunweiSUN commented 7 years ago

请问自己训练模型的时候到了这一步 过了一段时间进程被kill掉了 原因是内存不足吗？

hzy46 commented 7 years ago

可能是的。可以监控一下内存使用情况

JunweiSUN commented 7 years ago

还想再请教您一个问题 就是我在电脑中开了虚拟机来训练wave.yml，迭代到1000的时候有了一个wave.ckpt-1000 然后我把进程杀掉了，发现用这个迁移的效果比云盘里的wave.ckpt-done差了不少，请问要生成ckpt-done需要迭代多少次？还有就是迭代1000次之后他告诉我我用的是tensorflow V1，之后会切换到tensorflow V2，升级tensorflow是不是能解决这个问题？谢谢您啦。