求助，训练遮罩模型报错

肘王

星级打分

• 1
• 2
• 3
• 4
• 5

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

Error: 'pretrain'
Traceback (most recent call last):
  File "<frozen mainscripts.Trainer>", line 46, in trainerThread
  File "<frozen models.Model_XSeg.Model>", line 17, in __init__
  File "<frozen models.ModelBase>", line 238, in __init__
  File "<frozen models.ModelBase>", line 620, in get_summary_text
KeyError: 'pretrain'

RT，用的是猫之汉化佬的六一版本

lknet

https://fanyi.baidu.com
KeyError: 'pretrain'
预训练打开了吧，关上试试。

肘王

lknet 发表于 2024-8-7 21:59
https://fanyi.baidu.com
KeyError: 'pretrain'
预训练打开了吧，关上试试。

正在启动训练程序...



首次运行模型.

选择一个或多个 GPU 索引（用逗号分隔）.

  [0] :  DML: NVIDIA GeForce RTX 3060

[0] 选择哪些 GPU 索引? :
0


[wf] 脸型 ( h/mf/f/wf/head  按英文?获取帮助信息 ) : head
head
[8] 批量大小 ( 2-16  按英文?获取帮助信息 ) :
8
[n] 启用预训练模式 ( y/n ) :
n
加载样本中...: 100%|################################################################| 820/820 [00:02<00:00, 388.47it/s]
加载样本中...: 100%|################################################################| 789/789 [00:01<00:00, 415.15it/s]
Filtering: 100%|##################################################################| 1609/1609 [00:02<00:00, 640.68it/s]
使用 17 分割样本.
Error: 'pretrain'
Traceback (most recent call last):
  File "<frozen mainscripts.Trainer>", line 46, in trainerThread
  File "<frozen models.Model_XSeg.Model>", line 17, in __init__
  File "<frozen models.ModelBase>", line 238, in __init__
  File "<frozen models.ModelBase>", line 620, in get_summary_text
KeyError: 'pretrain'

感谢大佬回复！启用预训练模式那里一开始就是n，是要在其他地方关吗？

wtxx8888

肘王 发表于 2024-8-7 22:03
正在启动训练程序...

下载这个，六一的别看最新，不好。

https://dfldata.cc/forum.php?mod=viewthread&tid=5327