DFD224F神丹模型基础上继续训练100W+的杨幂专人模型

思念ii

深夜取经的路过～

令郎冲

支持一下~

panchunhe

牛逼！！！

fanyelaile

选择一个模型, 或者输入一个名称去新建模型.
[r] : 重命名
[d] : 删除

[0] : DFD224F - latest
:
0
加载名为 DFD224F_SAEHD 的模型...

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

  [0] :  DML: NVIDIA GeForce RTX 3070
  [1] :  DML: NVIDIA GeForce RTX 3070

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

Initializing models: 100%|###############################################################| 4/4 [00:00<00:00,  7.89it/s]
Traceback (most recent call last):
  File "<frozen mainscripts.Merger>", line 48, in main
  File "<frozen models.ModelBase>", line 238, in __init__
  File "<frozen models.ModelBase>", line 669, in get_summary_text
KeyError: 'random_src_flip'

完成鸟哇.
请按任意键继续. . .



楼主 这是怎么回事呢

xxw1991

有新得吗？

augensterns

fanyelaile 发表于 2024-6-25 01:48
选择一个模型, 或者输入一个名称去新建模型.
[r] : 重命名
[d] : 删除

修复一下dat文件就行：
import pickle

# 读取模型配置文件
with open("D:/你的路径/DFD224F_SAEHD_data.dat", "rb") as f:
    data = pickle.load(f)

options = data["options"]

# 添加默认缺失项（一次性补全）
defaults = {
    "random_src_flip": False,
    "random_dst_flip": False,
    "random_warp": True,
    "src_random_rotate": False,
    "src_random_scaling": False,
    "src_random_shift": False,
    "src_random_blur": False,
    "src_random_noise": False,
    "src_random_color": False,
    "src_random_downscale": False,
    "src_face_scale": 1.0,
}

for key, default_value in defaults.items():
    if key not in options:
        print(f"添加缺失项: {key} = {default_value}")
        options[key] = default_value

# 保存回去
with open("D:/你的路径/DFD224F_SAEHD_data.dat", "wb") as f:
    pickle.dump(data, f)

1308096098

好吧好吧