复现UNIT GAN的一些tips

ChengTT0607

于 2020-09-03 09:40:46 发布

阅读量715

点赞数 2

CC 4.0 BY-SA版权

文章标签：深度学习

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/GerinT/article/details/108375671

本文档介绍了复现UNIT GAN过程中的一些关键点，包括UNIT的原始论文、源码链接以及遇到的问题与解决方案。在尝试运行源码时，需要注意Python版本需为3.6，因为2.7版本会出现torch.utils相关错误。在Windows环境下，可能会遇到多线程问题导致的RuntimeError，尽管网上建议将multiprocessing部分放入main函数中，但在UNIT代码中未找到相关设置。因此，推荐使用Linux环境及Python 3.6来确保顺利运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

UNIT原文是英伟达在2017年NIPS上发表的关于scene translation的GAN：

Ming-Yu Liu, Thomas Breuel, Jan Kautz, “Unsupervised Image-to-Image Translation Networks” NIPS 2017 Spotlight, arXiv:1703.00848 2017
https://arxiv.org/pdf/1703.00848.pdf

UNIT源码链接如下：
https://github.com/mingyuliutw/UNIT
其中也有一个比较详细的教程

教程上python写的2.7和3.6，但是实际需要使用3.6才能跑通。
如果用2.7的话，会报关于torch.utils的若干错误，包括一个无法加载DataLoader的问题。
在Windows上运行的时候会遇到一个多线程的问题
RuntimeError: An attempt has been made to start a new process before the current process has finished its bootstrapping phase.
网上的链接都是说multiprocess的部分放在main函数里面就可以解决，if __name__ == '__main__':总体结构类似如下。但是实际上UNIT代码里面并没有找到这个多线程的设置，所以也不好改。

最低0.47元/天解锁文章

200万优质内容无限畅学

评论 1

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。