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

redorangetooth

星级打分

• 1
• 2
• 3
• 4
• 5

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

以汉化版、GTX10.RTX20版本为例。
众所周知，训练模型时可以设置“几小时备份一次”，但是这个选项默认只能是整数，无法做到1个小时以内的备份间隔。
并且这个选项并不意味着真正的保存时间间隔，真正的保存时间间隔在汉化版本里是25分钟（save_interval_min = 25）。
汉化版每隔25分钟保存并替换当前的模型文件，然后检查模型文件是否需要备份。如果你的备份时间是1个小时，那么每2到3次保存以后，会备份一个版本。
如果我们希望保存和备份同时进行，并且以分钟控制，效果如下：

则需要修改两个文件的代码：
文件一：ModelBase.py
这是我的文件位置，你们的根据自己的相对位置来找（F:\GTX10.RTX20\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\models\ModelBase.py）


文件一需要修改的代码部分如下：
第一处，代码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")

复制代码

第二处，代码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:

复制代码

文件二：Trainer.py
这是我的文件位置，你们的根据自己的相对位置来找（F:\GTX10.RTX20\DFL_maozhihanhua_RTX2080Ti\_internal\DeepFaceLab\mainscripts\Trainer.py）

文件二需要修改的代码部分如下：
删除文件35行的

1. save_interval_min = 25

复制代码

然后在is_reached_goal = model.is_reached_iter_goal()这一行上面紧挨着的地方添加一行

1. save_interval_min = model.autobackup_hour * 60

复制代码

实战效果如下，设置备份时间0.1小时，即每隔6分钟保存并且备份一次：



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

2648682296

感谢分享