EnlightenGAN 开源代码运行问题汇总

参考链接:

  1. EnlightenGAN 开源代码运行
  2. EnlightenGAN的运行环境搭建和训练自己的数据

源码下载和环境配置比较简单,本文测试环境:Win10 + RTX3060、cuda 11.3、python 3.8 + torch 1.12.0 + numpy 1.20.1

如果想修改在项目里创建test_daatset文件
(1) 将script.py文件中, ../final_dataset 更改为 final_dataset,然后在自己根目录下(其实就是文件夹EnlightenGAN中)创建一个文件夹 final_dataset,然后把数据trainA和trainB放入到里面,就是文中提到的real_A和real_B。(具体数据和模型,原作者github下有链接)在根目录下建一个model文件夹,把vgg的模型放到里面就很好了。
(2)script.py文件中将–gpu_ids 0,1,2 更改为 --gpu_ids 0,如果你是一个GPU就是这么改,要不然报错。
在这里插入图片描述
可能报错cuda内存不够,可以少几张照片运行:

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值