新建模型的预训练问题

fghfdg

星级打分

• 1
• 2
• 3
• 4
• 5

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

各位试过预训练的大佬，为什么我在pretrain_faces里面放了faceset.pak文件，新建一个模型打开预训练之后还是说报错啊。加载完模型准备加载头像的时候报错说没有文件。


现在只能直接在dst和src里面放头像文件，不开预训练选项了。

wtxx8888

文本编辑器，打开训练命令。

@echo off
call _internal\setenv.bat

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

pause

蓝色的，代表预训练数据的位置。可以自己改

fghfdg

wtxx8888 发表于 2024-12-23 19:55
文本编辑器，打开训练命令。

@echo off

感谢大佬

fghfdg

wtxx8888 发表于 2024-12-23 19:55
文本编辑器，打开训练命令。

@echo off

初始化模型...: 100%|###########################################################################################################################################################################################################################| 5/5 [00:05<00:00,  1.08s/it]
加载样本中...: 0it [00:00, ?it/s]
Error: 未提供训练数据.
Traceback (most recent call last):
  File "I:\DFL0504_DirectX12\_internal\DeepFaceLab\mainscripts\Trainer.py", line 59, in trainerThread
    debug=debug)
  File "I:\DFL0504_DirectX12\_internal\DeepFaceLab\models\ModelBase.py", line 205, in __init__
    self.on_initialize()
  File "I:\DFL0504_DirectX12\_internal\DeepFaceLab\models\Model_SAEHD\Model.py", line 860, in on_initialize
    generators_count=src_generators_count
  File "I:\DFL0504_DirectX12\_internal\DeepFaceLab\samplelib\SampleGeneratorFace.py", line 49, in __init__
    raise ValueError('未提供训练数据.')
ValueError: 未提供训练数据.

fghfdg

wtxx8888 发表于 2024-12-23 19:55
文本编辑器，打开训练命令。

@echo off

才发现原先也不是pretrain_faces而是pretrain_CelebA，我改为data_src\aligned了改完还是那个个报错啊。

wtxx8888

fghfdg 发表于 2024-12-23 20:21
才发现原先也不是pretrain_faces而是pretrain_CelebA，我改为data_src\aligned了改完还是那个个报错啊。 ...

要是SRC的话，还要改前面百分号内的%INTERNAL%，要改成，跟上面SRC一样%WORKSPACE%。

fghfdg

wtxx8888 发表于 2024-12-23 20:49
要是SRC的话，还要改前面百分号内的%INTERNAL%，要改成，跟上面SRC一样%WORKSPACE%。 ...

@echo off
call _internal\setenv.bat

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

pause
没懂，这是现在的bat，大佬我应该怎么改

fghfdg

fghfdg 发表于 2024-12-23 20:52
@echo off
call _internal\setenv.bat

哦，我知道了

wtxx8888

fghfdg 发表于 2024-12-23 20:52
@echo off
call _internal\setenv.bat

@echo off
call _internal\setenv.bat

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

pause

蓝色，换成紫色的，预训练项目，用的数据就是SRC。
我就不吐槽，你的理解力了。
光改DATA，前面百分号内的，上级目录不改，它能对，都出神了。

fghfdg

wtxx8888 发表于 2024-12-23 20:57
@echo off
call _internal\setenv.bat

感谢大佬，问题解决了