search

csjliang / LPTN

Official implementation of the paper 'High-Resolution Photorealistic Image Translation in Real-Time: A Laplacian Pyramid Translation Network' in CVPR 2021
Apache License 2.0
413 stars 47 forks source link

关于训练问题 #47

Open Maozhiquan opened 8 months ago

Maozhiquan commented 8 months ago

非常感谢您宝贵的工作！我有几个问题想请教您！

  1. 使用了您的代码（无更改）进行训练时，我发现训练100个iters时居然要25分钟，gpu占用内存才5个g，我认为这是不太合理的，因为您论文里面说了高效，我认为时间不应该那么长才对，我是哪里出了问题吗？？？我是单卡 一个batchsize为32，数据就800张
  2. 您的代码训练时应该可以边训练边验证，可是我在visualization并没有看到保存的生成的图片，val参数下save_img已经设置为true，文件夹里有每一轮的文件夹 可是里面没有图片，我感觉很奇怪！
  3. 我对您的代码进行修改后进行训练，发现psnr总是跳来跳去 一下20 一下22这种很奇怪，我想请问这是什么原因呢，因为我加了一个损失 所以最终各损失权重不是您论文里的，我个人认为是权重问题，但是我并不知道该如何设置，想请教一下您 最后，非常非常感谢您的伟大的工作，如果您能回答我，我将感激不尽！！！！！！！！！！！！！