XieMc

+3

> 已经修改，重新复制一下train文件的内容 老哥，我看你把model.load_state_dict(pretrained_dict)改为model.load_state_dict(model_dict)，但似乎还是报这个错，除非把config.py的num_classes改回21就不报错了，另外，大神能留个联系方式吗

> 有一句忘了复制了 > pretrained_dict = {k: v for k, v in pretrained_dict.items() if np.shape(model_dict[k]) == np.shape(v)} 谢谢，现在train.py的训练没有问题，但是predict.py预测仍然报错如下： Using TensorFlow backend. 2020-03-26 14:51:12.004041: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'cudart64_100.dll';...