【求助】liae结构的迪丽热巴万能模型训练出错

dfllearner · 发表于 2023-7-19 00:02:38

ypfriend 发表于 2023-7-18 23:44
感谢大佬，把 models_opt_on_gpu关了后，其他参数仍旧默认，就可以跑起来了，看了任务管理器，显存只用了 ...

ae_dims，这个维度参数改不了，在创建模型时就定了
gan_dims，这个GAN维度可以改

ypfriend · 发表于 2023-7-19 00:37:08

dfllearner 发表于 2023-7-19 00:02
ae_dims，这个维度参数改不了，在创建模型时就定了
gan_dims，这个GAN维度可以改 ...

好的，感谢感谢，我再慢慢研究

seancai110 · 发表于 2023-7-19 02:19:07

ypfriend 发表于 2023-7-18 23:46
感谢大佬，关掉GPU参数就可以跑了，但是gan维度一般是多少好呢

改16可以了。有神经网络经验的都知道，一般鉴别器比生成器跑得快，所以鉴别器没必要开这么大维度。

ypfriend · 发表于 2023-7-19 19:49:05

seancai110 发表于 2023-7-19 02:19
改16可以了。有神经网络经验的都知道，一般鉴别器比生成器跑得快，所以鉴别器没必要开这么大维度。 ...

感谢感谢大佬

er232 · 发表于 2023-10-20 15:00:37

学习学习

		自动登录	找回密码
密码			立即注册（仅限QQ邮箱）

【求助】liae结构的迪丽热巴万能模型 训练出错