我的训练环境:
Windows10 64bit;
MindSpore1.3.0;
CPU;
python3.7.5;
【操作步骤&问题现象】
在使用load_checkpoint函数加载已经保存的模型时,提示有几个参数无法成功载入。
【截图信息】
resnet50的预训练模型是用imagenet数据集的, 一共1000类,如果用猫狗分类,一共2类,所以最后的fc层是加载不进去的
我的训练环境:
Windows10 64bit;
MindSpore1.3.0;
CPU;
python3.7.5;
【操作步骤&问题现象】
在使用load_checkpoint函数加载已经保存的模型时,提示有几个参数无法成功载入。
【截图信息】
resnet50的预训练模型是用imagenet数据集的, 一共1000类,如果用猫狗分类,一共2类,所以最后的fc层是加载不进去的