lada模型训练要求太高了，显存需要24G+

androw01 · 发表于 2026-1-8 18:10:04

云端GPU训练mosaic-restoration-basicvsrpp模型，显存直接占用20G+，
数据集还只是一个5分钟的视频

，速度也很慢，先练练手,希望没炼错

androw01 · 发表于 2026-1-8 18:43:18

本帖最后由 androw01 于 2026-1-14 21:20 编辑

看来要芭比q了，第二步转换trained weights into the GAN-compatible model报错

========================================
看了mmengine训练文档，开启AMP、bfloat16，BS设置为1，
显存训练显存可以控制在10G+左右，速度也会增加

androw01 · 发表于 2026-1-8 19:25:12

转换完成，到了训练第三步，显存直接占用24G+，恐怖如斯

androw01 · 发表于 2026-1-8 21:25:00

已放弃，步骤太繁琐，坑太多，，还是更喜欢炼超分Real-Esrgan模型

wangsh · 发表于 2026-1-8 21:57:15

本帖最后由 wangsh 于 2026-1-8 21:58 编辑

楼主牛，有没有练好的Real-Esrgan 2x 或 3x模型分享？

Zsural · 发表于 2026-1-9 08:45:42

前排围观，感谢分享

black-sky · 发表于 2026-1-9 12:33:11

看起來好高端這是把模組拿到雲端去操作的意思嗎?

devidlu · 发表于 2026-1-9 20:56:28

新人，虽然看不懂，不明觉厉

ol580095004 · 发表于 2026-1-10 03:39:03

不然呢。。。你以为呢。。。训练模型又不是部署模型进行推理，即便是可以部署在嵌入式设备的模型，训练的时候都会吃掉几十G显存，理论上几乎是没有上限的，有多少显存都可以通过训练时候的参数给你吃满。

不兼容 · 发表于 2026-1-10 09:25:12

24g的显存，用5090的8卡机基本差不多吧。还是要H200？

		自动登录	找回密码
密码			立即注册