【新版尝鲜】MVE Pytorch trainer GUI

青青子衿 · 发表于 2024-10-10 14:20:03

本帖最后由青青子衿于 2024-10-12 10:25 编辑

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

july · 发表于 2024-10-10 16:18:30

刚看了iperov的主页，居然删库了。
这是国外大佬开发的吗？看来国外玩家还是很狂热啊

青青子衿 · 发表于 2024-10-10 16:51:25

july 发表于 2024-10-10 16:18
刚看了iperov的主页，居然删库了。
这是国外大佬开发的吗？看来国外玩家还是很狂热啊 ...

MVE团队搞得pytorch版本，正在试验

july · 发表于 2024-10-11 13:09:02

青青子衿发表于 2024-10-10 16:51
MVE团队搞得pytorch版本，正在试验

兼容原版模型吗？

青青子衿 · 发表于 2024-10-11 13:11:23

july 发表于 2024-10-11 13:09
兼容原版模型吗？

兼容原版，mve也支持，其他的没试过

july · 发表于 2024-10-11 20:21:15

青青子衿发表于 2024-10-11 13:11
兼容原版，mve也支持，其他的没试过

回头试试

jinyuan37 · 发表于 2024-10-23 09:07:36

重写主要是为了版权问题吧

[DFL版本] 【新版尝鲜】MVE Pytorch trainer GUI

评分

热心会员

突出贡献