search

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
https://cg.cs.tsinghua.edu.cn/jittor/
Apache License 2.0
3.08k stars 311 forks source link

model.load()加载预训练模型失败 #375

Open wszhengjx opened 2 years ago

wszhengjx commented 2 years ago

image

diamond0910 commented 2 years ago

您好，检查下你的模型参数名称是否一致？

wszhengjx commented 2 years ago

是一致的，关键在于这个1 params，我怀疑他是将load("/home/xxx/.......pth")中的"/home/xxx/.......pth"整体当作一个参数了，请问为什么会出现这种情况？jittor不能从本地加载pth文件嘛？