deepfacelab中文网

 找回密码
 立即注册(仅限QQ邮箱)
查看: 804|回复: 1

自定义模型训练时的备份时间间隔的方法

[复制链接]

2

主题

11

帖子

132

积分

高级丹童

Rank: 2

积分
132

开心娱乐节日勋章

 楼主| 发表于 2023-9-3 16:13:12 | 显示全部楼层 |阅读模式
星级打分
  • 1
  • 2
  • 3
  • 4
  • 5
平均分:NAN  参与人数:0  我的评分:未评
本帖最后由 redorangetooth 于 2023-9-3 16:26 编辑

以汉化版、GTX10.RTX20版本为例。
众所周知,训练模型时可以设置“几小时备份一次”,但是这个选项默认只能是整数,无法做到1个小时以内的备份间隔。
并且这个选项并不意味着真正的保存时间间隔,真正的保存时间间隔在汉化版本里是25分钟(save_interval_min = 25)。
汉化版每隔25分钟保存并替换当前的模型文件,然后检查模型文件是否需要备份。如果你的备份时间是1个小时,那么每2到3次保存以后,会备份一个版本。
如果我们希望保存和备份同时进行,并且以分钟控制,效果如下:
image.png
则需要修改两个文件的代码:
文件一:ModelBase.py
这是我的文件位置,你们的根据自己的相对位置来找(F:\GTX10.RTX20\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\models\ModelBase.py
image.png

文件一需要修改的代码部分如下:
第一处,代码291行的

  1. self.options['autobackup_hour'] = io.input_int(f"几个小时备份一次", default_autobackup_hour, add_info="0..24", help_message="自动备份模型文件,每 N 小时预览一次。 最新备份位于 model/<>_autobackups/01")
复制代码

修改为
  1. self.options['autobackup_hour'] = io.input_number(f"几个小时备份一次", default_autobackup_hour, add_info="0..24", help_message="自动备份模型文件,每 N 小时预览一次。 最新备份位于 model/<>_autobackups/01")
复制代码

image.png
第二处,代码411行的
  1. diff_hour = int ( (time.time() - self.autobackup_start_time) // 3600 )
复制代码

修改为
  1. diff_hour =  (time.time() - self.autobackup_start_time) / 3600
复制代码

第三处,代码413行的
  1. if diff_hour > 0 and diff_hour % self.autobackup_hour == 0:
复制代码

修改为
  1. if diff_hour >= self.autobackup_hour:
复制代码

image.png

文件二:Trainer.py
这是我的文件位置,你们的根据自己的相对位置来找(F:\GTX10.RTX20\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\mainscripts\Trainer.py
image.png
文件二需要修改的代码部分如下:
删除文件35行的
  1. save_interval_min = 25
复制代码
然后在is_reached_goal = model.is_reached_iter_goal()这一行上面紧挨着的地方添加一行
  1. save_interval_min = model.autobackup_hour * 60
复制代码

image.png

实战效果如下,设置备份时间0.1小时,即每隔6分钟保存并且备份一次:
image.png


代码修改的方式已经给出,如果不想自己修改,那就买网盘里改好的文件,并且覆盖原位置的文件即可。

image.png
回复

使用道具 举报

0

主题

23

帖子

297

积分

初级丹师

Rank: 3Rank: 3

积分
297

万事如意节日勋章

发表于 2024-1-30 19:31:31 | 显示全部楼层
感谢分享
回复

使用道具 举报

QQ|Archiver|手机版|deepfacelab中文网 |网站地图

GMT+8, 2024-11-12 09:16 , Processed in 0.122849 second(s), 35 queries .

Powered by Discuz! X3.4

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表