DeepFaceLab_NVIDIA_RTX5000_RTX4000 Pytorch个人免费版1.5.2 12月25日更新

小王子

星级打分

• 1
• 2
• 3
• 4
• 5

平均分:5  参与人数:1  我的评分:未评

更新：
2025年12月5日更新Trainer 1.5.2版本更新内容

- 当内存发生变化、图形更新或冻结状态更新时，优化器会被重新加载
- 优化器更新后，训练器的VRAM内存会被清理
- 修复了掩码模型未完全冻结的问题
- 修复了第一次迭代崩溃的问题（希望如此）
- 目前已移除部分float16

11月5日更新Trainer 1.5.1版本

- 用于设置训练的图形用户界面（GUI），将会舍弃配置文件逐步转为启动时在线实时修改参数
- 修复了启用LRD的GAN的稳定性问题
- 修复ICE训练器兼容性问题
- 增加了梯度检查点
- 在图形用户界面（GUI）中添加了配置组名称
- 在模型设置完全加载之前禁用设置菜单 - 仅在运行服务器时


购买过的直接下载torch_trainer-1.5.1.1-noenv.zip，重命名备份torch_trainer文件夹后把压缩包里的torch_trainer文件夹放到对应的位置，然后用备份的saehd_config.yaml或amp_config.yaml替换掉新的saehd_config.yaml或amp_config.yaml完成更新。

因为TensorFlow‌‌2.11版本开始‌，停止了对Windows系统原生CUDA构建的支持，JesterX和他的团队对DFL把tensorflow替换为pytorch进行了重写，当前最新版本为DFL Pytorch Trainer Version 1.4.2，有JesterX的discord可以在线进行支持，发新版后本帖持续更新。

Pytorch版本原生支持NVIDIA_RTX5000和RTX4000显卡，大大提高RTX5000和RTX4000系列显卡训练速度并且减少一部分显存使用量，RTX3000显卡理论上新版本的CUDA会提升一些。
测试使用笔记本4070显卡，之前batch_size设为4跑不起来，现在batch_size设为6并且开Gan可以正常跑。

兼容原版和MVE分支
新版特征：
移植DFL原版的训练模式和配置
新的训练WEB GUI
示例历史记录浏览器
可修改的实时设置
其他损失信息

数据集缓存/自动提取到模型分辨率

- 修复了导入预训练模型加载失败的形状错误
- 修复了AMP DFM合并问题（位于DFL DFM合并分支内）
- 修正了编码器层的模型配置顺序（内部层编号为0，分辨率较高的层编号最大）

个人免费版最大分辨率max resolution提高为448，收费版本不限，linux版本和windows收费版本需要购买，1年的服务版2100。

使用方法：
1.训练模型和src和dst训练素材位置
和DFL原版一样，把训练模型放在workspace文件夹下的model的文件夹中，src和dst训练素材分别放在data_src和data_dst中，注意：不要放pak文件，要解包出来

2：生成和修改配置文件
注意：生成配置文件之前遮罩模型先不要放入model的文件夹
模型和素材都放好后，如果是SAEHD的模型运行export_saehd_config.bat脚本生成配置文件saehd_config.yaml，如果是amp模型运行export_amp_config.bat脚本生成配置文件amp_config.yaml，然后使用文本编辑器打开配置文件，打开后是这样的内容：




里面的是他配置不要改，根据需要把color_augmentation(对应的是cc-aug)参数改成false，默认为true
autobackup_hour自动备份时间根据需要修改一下。

3.训练
运行train_saehd.bat脚本开始训练，cmd加载窗口


加载过程中会把素材压缩为和模型分辨率相应的大小，等加载完且无报错的情况下复制如下代码浏览器打开
http://localhost/
然后即可看到如下训练界面


模型参数在线配置界面
有些可以在线改，不能在线修改的需要退出后在saehd_config.yaml中修改。


点击Open Tensorboard打开tb界面可以看曲线图


中止训练退出，可以在WEB界面的点击Save或Exit

4. 合成图片：
最后加入遮罩等操作后，运行simple_merge_saehd.bat合成图片

别看我我不在

报错了，用不了
'F:\DFL' 不是内部或外部命令，也不是可运行的程序
或批处理文件。
请按任意键继续. . .

azrael

还有就是弱弱的问下，为什么我运行export_saehd_config.bat和train_saehd.bat的时候显示不是内部或外部命令，也不是可运行的程序或批处理文件

july

小王子 发表于 2025-10-2 00:11
问开发者了，没有增加A卡的代码，nvidia cards are only supported，目前为止只支持N卡 ...

谢谢
discord是付费进的吗？

seancai110

pytorch理论上支持a卡和i卡吧

july

这个版本训练好的模型能在原版DFL合成吗？

liyungui88

有中文版的吗，这个版本

小王子

seancai110 发表于 2025-10-1 19:23
pytorch理论上支持a卡和i卡吧

问开发者了，没有增加A卡的代码，nvidia cards are only supported，目前为止只支持N卡

小王子

july 发表于 2025-10-1 23:08
这个版本训练好的模型能在原版DFL合成吗？

可以，模型完全通用。不过你要是新显卡还是在这版本合成比较快

小王子

liyungui88 发表于 2025-10-1 23:52
有中文版的吗，这个版本

没中文的，英文的用几次就习惯了，就是改改saehd_config.yaml中的参数，没几个主要单词，相应的翻译记个笔记就好了。webui可以用在线翻译

seancai110

小王子 发表于 2025-10-2 00:11
问开发者了，没有增加A卡的代码，nvidia cards are only supported，目前为止只支持N卡 ...

晕倒，pytorch的跨平台搞了这么久，我还以为现在pytorch可以很简单的用在a卡上了。

ajno22

请问楼主，有夸克网盘吗？