运行e4e+StyleCLIP Colab代码出现net未定义及UnpicklingError报错
报错问题排查及解决方法
错误1:NameError: name 'net' is not defined
产生原因
- 代码块运行顺序错误:推理代码块依赖的
net变量由模型加载代码块定义,你跳过了模型加载步骤直接运行推理代码,变量未被创建 - 模型加载代码块运行时报错终止,
net变量未成功初始化
解决方法
- 严格按照Colab笔记本从上到下的顺序依次运行所有代码块,不得跳步执行
- 先确认模型加载代码块成功输出
Model successfully loaded!提示后,再运行后续的推理代码块
错误2:UnpicklingError: invalid load key, '<'
产生原因
torch.load()读取的不是合法的PyTorch序列化权重文件:
- 模型权重下载链接失效/重定向,实际下载到的是HTML格式的网页文件,文件首字符为HTML的
<标签触发报错 - 下载过程中网络中断,权重文件未下载完整导致损坏
解决方法
- 首先检查
EXPERIMENT_ARGS['model_path']对应的文件大小,若仅几KB则确认是下载了网页文件,直接删除该损坏文件 - 重新获取有效预训练权重下载地址,下载完整的pSp权重文件后上传到谷歌云盘对应路径,替换损坏文件
- 若使用脚本自动下载,可在下载完成后校验文件大小与官方公布的权重大小一致,再执行模型加载代码
内容的提问来源于stack exchange,提问作者Althaen
相关产品推荐
相关产品推荐

