1.最近在colab上把代码迁移到本地,在加载模型参数的时候出错
torch.load('../pkl/create_songs(lstm).pkl', map_location=device)
报错:
Can’t get attribute ‘_default_unk_index’ on <module 'torchtext.vocab’
百度搜不到结果,上谷歌查了一下有人提出版本问题,看了一下果然是,保存和加载的时候两者版本最好一致,不然真的坑死了!!!
解决方法:
conda install torchtext==xxx