stylegan2内存不足

by100

masker 发表于 2022-3-4 18:14
调小了bs值还是不行，会不会还有其他参数决定？

数据集的加载方式知道吗？
是直接逐个调用还是全部加载后逐个调用？

masker

by100 发表于 2022-3-4 18:19
数据集的加载方式知道吗？
是直接逐个调用还是全部加载后逐个调用？

不知道，刚接触还不太懂，刚看了一下显存和ARM都爆了，改bs，改图片分辨率都不行，
这是colab的代码：链接: https://pan.baidu.com/s/10YChj5Dx5WkCqyx5H67wpg?pwd=xsr7 提取码: xsr7
这是项目地址：https://github.com/NVlabs/stylegan2
大佬有空可以研究研究，谢谢

by100

masker 发表于 2022-3-4 19:09
不知道，刚接触还不太懂，刚看了一下显存和ARM都爆了，改bs，改图片分辨率都不行，
这是colab的代码：链 ...

可以考虑用stylegan3,
效果更好

masker

by100 发表于 2022-3-4 19:23
可以考虑用stylegan3,
效果更好

啊？有3了？我都不知道，我去看看

by100

masker 发表于 2022-3-4 19:29
啊？有3了？我都不知道，我去看看

https://github.com/NVlabs/stylegan3

masker

by100 发表于 2022-3-4 19:39
https://github.com/NVlabs/stylegan3

这玩意儿也太耗时了吧，训练了一个小时动都不动一下

xuxiao794613

不懂帮顶！

by100

masker 发表于 2022-3-4 20:45
这玩意儿也太耗时了吧，训练了一个小时动都不动一下

不会是从头练的？
用了GPU吗？什么显卡？

masker

by100 发表于 2022-3-7 11:33
不会是从头练的？
用了GPU吗？什么显卡？

是从头练的，tesla k80，难道可以像DFL一样用别人的模型接着练吗? 还有就是我发现他可能压根没在练，程序没有停，但是模型文件没有任何修改，修改时间还显示的我创建的时间。求大佬指点