【已解决】：mulan模型训练-错误提示【未提供训练数据】？

mclove · 发表于 2025-4-13 13:08:27

本帖最后由 mclove 于 2025-4-13 19:27 编辑

小白刚入门：

使用了【192DF-WF-6g可用-五彩艺术】进行 SAEHD模型训练、合成都正常。
但是使用【Mulan-DF256-V71】进行MULAN模型训练、合成都提示错误【未提供训练数据】？
特此求教，如何解决？谢谢！

具体如下：

载入训练模块：ICE-V71
使用设备: 0-NVIDIA GeForce RTX 5080 (12.88gb/12.88gb)
ICE训练器：请稍等，正在优化 GPU ...,

装载类型:me-model: 100%|██████████████████████████| 8/8 [00:05<00:00,  1.42it/s]
装载训练样本: 0it [00:00, ?it/s]
Traceback (most recent call last):
  File "models_ice\IceModel_Train.py", line 55, in models_ice.IceModel_Train.trainerThread
  File "models_ice\IceModel_Class.py", line 153, in models_ice.IceModel_Class.IceModel_Class.__init__
  File "models_ice\Ice_MULAN\Model.py", line 1482, in models_ice.Ice_MULAN.Model.MULANModel.on_init_model
  File "samplelib\SampleGeneratorFace.py", line 49, in samplelib.SampleGeneratorFace.SampleGeneratorFace.__init__
ValueError: 未提供训练数据，请检查素材是否放置。No training data provided.

mclove · 发表于 2025-4-13 13:13:43

使用SAEHD模型训练正常，【如下所示】：

模型摘要  ________________________

模型名称:: DF 192 wf_SAEHD

当前迭代: 17365

________________________  模型参数  ________________________

  模型架构: df
  人脸类型: wf
  模型分辨率: 192
  自动编码器维度 ae_dims: 384
  编码器维度 e_dims: 92
  解码器维度 d_dims: 72
  解码器掩码维度 d_mask_dims: 22
  BS大小: 8
  学习率lr: 5e-05
  遮罩约束训练范围: 开启
  均衡侧脸训练权重: 开启
  眼睛和嘴部优先: 开启
  素材增强-随机扭曲: 开启
  素材增强-翻转dst: 开启
  RG优化器: 开启
  信仰优化器 adabelief: 开启
  模型优化器放置于GPU: 开启
  梯度剪裁: 开启
  版本: V3

____________________________________________________________

  设备序号: 0
  设备名称: NVIDIA GeForce RTX 5080
  显存大小: 12.88GB
____________________________________________________________

ICE 1.818 version by kingboy! QQ group:366893641

开始训练，预览窗口按 "H" 键获得功能帮助。

[当前时间][i:迭代数量][延迟ms]-[src损失][dst损失]
[13:12:35][i:00017463][0406ms]-[0.37763][0.30647]

mclove · 发表于 2025-4-13 13:44:20

我在网上查到的解决方法。
但是，使用SEAHD模型是没有问题的啊？奇怪了...

问题15：no training data found
出现这个报错有点离大谱，你要训练起码得有素材吧，没有素材训练个卵？要是有素材，看你的文件目录名字是不是错的，正确名字应该是data_dst和data_src

mclove · 发表于 2025-4-13 19:26:33

解决办法找到了：
进入MULAN模型训练时，设置模型参数，关闭预训练就可以正常训练了！