DeepFaceLab_NVIDIA_RTX5000_RTX4000 Pytorch 1.4.2个人免费版

小王子

星级打分

• 1
• 2
• 3
• 4
• 5

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

因为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合成图片

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卡上了。

july

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

谢谢
discord是付费进的吗？

ajno22

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