请问8G显存的GTX1070显卡最适合的训练分辨率及参数是多少？

windjet

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:NAN  参与人数:0  我的评分:未评

如标题所示，预计要对应一下2K甚至4K视频，想知道尽可能提高清晰度的前提下，GTX1070显卡最适合的训练分辨率及参数是多少？当然至少是要训练整个脸WF。
以前一直训练WF256，DF模型结构，只能跑BS4，看AE architecture参数说明如下，

参考：
-D:这个版本旨在提高效率。同等配置下，它可以让你以两倍的分辨率训练你的模型，比如之前只能跑128，启用之后可以跑256。如果分辨率相同，单次迭代时间更少，消耗显存更少。但是在质量上可能会有所降低，有待验证。

理论上使用DF-D可以跑WF512，但实际上跑不起来，就是BS1也不行，不知道是什么原因，所以想问一下各位GTX1070显卡最适合的训练分辨率及参数是多少。

cjlzkl

用me版本。6g都能跑512

2772403575

1070就不要想着对标2K甚至4K了，256高参到320中高参已经是极限了

ceeyee

cjlzkl 发表于 2023-1-5 12:30
用me版本。6g都能跑512

请问可以给个链接吗？谢谢。
我在软件下载版，搜"me"，出来都是codefomer, chrome，等，没有找到me版DFL。

windjet

2772403575 发表于 2023-1-5 12:39
1070就不要想着对标2K甚至4K了，256高参到320中高参已经是极限了

这样啊，谢谢了，我是看参数说明好像能跑512的。

windjet

cjlzkl 发表于 2023-1-5 12:30
用me版本。6g都能跑512

谢谢提示，但什么是me版本啊？训练的模型以后能和官方版本兼容吗？另外为啥他就能比官方版本跑更高的分辨率呢？

cjlzkl

windjet 发表于 2023-1-5 13:45
谢谢提示，但什么是me版本啊？训练的模型以后能和官方版本兼容吗？另外为啥他就能比官方版本跑更高的分辨 ...

显存不够内存凑，你把模型放内存，关掉a啥的优化器。原版应该也能跑。me论坛自搜

cjlzkl

ceeyee 发表于 2023-1-5 12:51
请问可以给个链接吗？谢谢。
我在软件下载版，搜"me"，出来都是codefomer, chrome，等，没有找到me版DFL。 ...

第五个。。。。。。。。。。。。。。

Aufheben

me版和rg版似乎已经不能下载了。可惜。

ceeyee

cjlzkl 发表于 2023-1-5 14:00
第五个。。。。。。。。。。。。。。

看到了，谢谢。