训练自动保存时间修改成功

晓雷哥哥 · 发表于 2024-3-1 00:20:13

本帖最后由晓雷哥哥于 2024-3-1 18:44 编辑

因为我是A卡，训练很慢，啥都没优先没gan得2秒一次迭代

再加上每二十分钟自动保存会卡个几分钟，能修改20分钟自动保存吗？
后来一个一个文件py笔记本打开寻找终于找到了

\_internal\DFL\mainscripts\Trainer.py文件笔记本打开
找到

while True:
      try:
         start_time = time.time()

         save_interval_min = 25

         if not training_data_src_path.exists():
            training_data_src_path.mkdir(exist_ok=True, parents=True)

里面的save_interval_min就是自动保存时间默认是25

独孤求败 · 发表于 2024-3-1 04:32:09

假如你是训练 train SAEHD 的脚本运行
那么用记事本打开该脚本，以下红色字添加上去
其中60代表 60分钟自动存档一次,默认是25分钟

@echo off
call _internal\setenv.bat

"%PYTHON_EXECUTABLE%" "%DFL_ROOT%\main.py" train ^
--saving-time 60 ^
--training-data-src-dir "%WORKSPACE%\data_src\aligned" ^
--training-data-dst-dir "%WORKSPACE%\data_dst\aligned" ^
--pretraining-data-dir "%INTERNAL%\pretrain_CelebA" ^
--model-dir "%WORKSPACE%\model" ^
--model SAEHD

pause

晓雷哥哥 · 发表于 2024-3-1 16:23:49

独孤求败发表于 2024-3-1 04:32
假如你是训练 train SAEHD 的脚本运行
那么用记事本打开该脚本，以下红色字添加上去
其中60代表 60分钟自动 ...

多谢大佬，回去试试

晓雷哥哥 · 发表于 2024-3-1 17:51:44

独孤求败发表于 2024-3-1 04:32
假如你是训练 train SAEHD 的脚本运行
那么用记事本打开该脚本，以下红色字添加上去
其中60代表 60分钟自动 ...

usage: main.py [-h]
{extract,sort,util,train,exportdfm,merge,videoed,facesettool,dev_test,xseg}
...
main.py: error: unrecognized arguments: --saving-time 60

独孤求败 · 发表于 2024-3-1 23:49:27

晓雷哥哥发表于 2024-3-1 17:51
usage: main.py [-h]
{extract,sort,util,train,exportdfm,merge,videoed,facesettool,de ...

2024年02月14日，一条龙整套 DF 汉化版解压即用免安装，有需要者请自行下载
(内含新版 DeepFaceLive 实时换脸程序)
显卡RTX1050-2080下载连结
https://pan.baidu.com/s/1rzsz0Cv6HahcqGIf4fB6sQ?pwd=bvrx

显卡RTX3050-4090下载连结
https://pan.baidu.com/s/1FdXWENI-9TcApnu92UetTw?pwd=7d5l

训练自动保存时间修改成功

隐藏大神

突出贡献

论坛元老

咸鱼勋章