我有一个运行有PyTorch代码的Google Colab笔记本。 在训练功能的开始,我使用以下代码毫无问题地创建,保存和下载word_to_ix和tag_to_ix字典:
from google.colab import files
torch.save(tag_to_ix, pos_dict_path)
files.download(pos_dict_path)
torch.save(word_to_ix, word_dict_path)
files.download(word_dict_path)
我训练模型,然后尝试使用代码下载它:
torch.save(model.state_dict(), model_path)
files.download(model_path)
然后我得到一个MessageError: TypeError: Failed to fetch
。
显然,第三方cookie(建议here)没有问题,因为下载了第一个文件没有问题。 (实际上,我也尝试在我的Allow
部分中添加链接,但是,出乎意料的是,这没有什么区别。)
我最初试图按原样保存模型(据我了解,将其另存为Pickle),我认为Colab文件可能无法很好地下载Pickles,但是正如您在上面看到的,我m现在尝试保存一个dict对象(也是word_to_ix和tag_to_ix的对象),并且仍然无法正常工作。
用鼠标右键单击手动下载文件不是一个解决方案,因为有时候我在执行其他操作时仍使代码运行,而当我回到文件中时,运行时已断开,文件消失了
有什么建议吗?