【新版尝鲜】DFL pytorch trainer

luckXu · 发表于 2024-10-11 10:50:43

星级打分

1
2
3
4
5

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

本帖最后由 luckXu 于 2024-10-11 06:05 编辑

DFL Pytorch版本

JesterX和他的团队对DFL使用pytorch进行了重写，这个版本说的是从去年开始到最近才发出来尝鲜版
兼容原版和MVE分支
新版特征：
特征：

移植的 DFL 训练模式和配置

新的训练 GUI

示例历史记录浏览器

可修改的实时设置

其他损失信息

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

附加增强：将 SRC 数据集添加到 LIAE 模型的 DST IN 中

多种训练精度模式（full、full with tensor cores、float16 和 bfloat）

现在发出来的时个人版，个人版支持的最大分辨率为384

正片开始：

1.目录结构

如果你训练的是SAEHD模型，和DFL原版一样，在workspace里面放入相应的训练集和模型，注意：不要放pak文件，要解包出来

然后把这个dll文件解压出来放到：C:\Windows\System32 文件夹里面

不然会报如下错误

2：删除和修改配置文件
模型和素材都放好后，把根目录下的saehd_config.yaml文件删除，这个文件是默认的，和放置的模型不匹配
然后运行export_saehd_config.bat脚本，重新生成配置文件，然后使用文本编辑器打开配置文件，打开后是这样的界面

里面的是他配置不要改，个人喜欢把下面的参数改一下

把这个颜色转换改为false，这个颜色参数是基于cc-aug修改而而来，可以到后期修改为true

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

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

模型配置界面

点击Open Tensorboard打开tb界面

在训练过程中可以点击预览图片或者Update Preview按钮更新图片

关于训练速度，有一定的提升，更多的还没来得及测试。

可以下载这两个文件尝鲜，链接限时30天。

下载链接：https://pan.baidu.com/s/1FKewM-_5HdoN71sbp3CWsA

Lau9 · 发表于 2024-10-12 18:30:53

wtxx8888 发表于 2024-10-12 00:38
都喜欢搞自己的，但原版用户，始终是最多的。。。

百花齐放是好的，但如果闭源我就不会用

DFL小白02 · 发表于 2024-10-11 11:19:34

JesterX是谁

Cxsmo · 发表于 2024-10-11 12:15:23

it's is missing features that mve and stuff has and has Res limit if not paying and if you are going to pay xfacekit is a lot better and also speed increase from pytorch mve is not as much as they were acting like it was