windows 下 non-stationary_texture_syn_master 问题与解决

ubuntu 下non-stationary_texture_syn_master  跑这个项目是完全没问题的,主要在于

window下一对问题

1.一开始就跑不通

解决: 在 train 中添加以上代码块

2.在训练的过程中内存outofmemory

找到原因了,保存新生成的图像的时候,会占用GPU缓存,结果就不够用了,用torch.cuda.empty_cache()指令就不会让显存占用多余的无用空间了

猜你喜欢

转载自blog.csdn.net/Willen_/article/details/85858486