处理 GAN 训练输出的最佳方法是什么?

时间:2021-01-08 23:10:55

标签: python tensorflow jupyter-notebook pytorch generative-adversarial-network

监督 GAN 的训练通常不仅涉及输出指标,还包括以特定的 epoch 间隔输出图像。我的应用程序还涉及打印表格。我使用 jupyter 笔记本,但仅将其全部打印在笔记本上会使每个实验方式的每个笔记本过大(+100 MB),并且互联网浏览器因此变慢并经常崩溃。

我想通常的做法是将图像输出保存在其他地方(使用张量板或纯图像文件),但这对我来说并不理想,因为我喜欢将每个图像与文本/表格输出一起观察相对于它的时代。如果我能将整个训练输出保存到一个文件中会更好,这样我就可以向下滚动它,用文本/表格/图像观察每个 epoch 输出,就像 jupyter notebook 输出一样。有没有办法实现这个?或者也许是我没有考虑的更好的方法?

如果这个问题有什么不足之处,谢谢并抱歉。如果是这样,请告诉我,我会删除它。

1 个答案:

答案 0 :(得分:1)

嗯,这是一个偏好问题,有人喜欢如何拥有它。当我训练 GAN 时,我按以下方式处理它,对于损失和其他类似的值,我按照每个时期的方式将它们简单地打印在笔记本上,就像我们对任何其他模型所做的那样,我将生成由{% for i, j in items | zip(teams) %} {% set a = i + j %} {% endfor %} 并将其保存到文件夹中以检查生成器是否变得更好。但是为了让我更容易观察这些图像,我对这些图像使用了有意义的标签,我在标签中包含了纪元编号,这有助于快速浏览图像。然而,浏览图像会有点乏味,所以为了让它更舒服一点,我合并了每个时代(比如 500)产生的所有图片并制作了一个视频,观看起来非常酷。我发现这种训练 GAN 并生成其输出的方式非常容易,但正如我所说,这将是一个偏好问题。

这是代码- https://github.com/khalidsaifullaah/Classic-Deep-Learning-Models/tree/master/GAN