我想在google colab上将我的模型保存到我的驱动器中。我使用以下代码来安装我的Google云端硬盘 -
from google.colab import drive
drive.mount('/content/gdrive')
在完成所有预处理,模型定义和培训之后,我想将模型保存到驱动器中,因为培训需要很长时间。因此,我将保存它以定期驱动并从该点重新加载以继续。保存模型的代码是:
def save_model(model, model_name, iter):
path = f'content/gdrive/My Drive/Machine Learning Models/kaggle_jigsaw_{model_name}_iter_{iter}.pth'
print(f'Saving {model_name} model...')
torch.save(model.state_dict(), path)
print(f'{model_name} saved successfully.')
EMBEDDING_DIMS = 128
HIDDEN_SIZE = 256
gru = GRU(vocab.n_words, EMBEDDING_DIMS, HIDDEN_SIZE, 2).to(device)
save_model(gru, 'gru', 0)
我收到以下错误:
Saving gru model...
---------------------------------------------------------------------------
FileNotFoundError Traceback (most recent call last)
<ipython-input-27-d2510611a9d4> in <module>()
9
10 gru = GRU(vocab.n_words, EMBEDDING_DIMS, HIDDEN_SIZE, 2).to(device)
---> 11 save_model(gru, 'gru', 0)
<ipython-input-27-d2510611a9d4> in save_model(model, model_name, iter)
2 path = f'content/gdrive/My Drive/Machine Learning Models/kaggle_jigsaw_{model_name}_iter_{iter}.pth'
3 print(f'Saving {model_name} model...')
----> 4 torch.save(model.state_dict(), path)
5 print(f'{model_name} saved successfully.')
6
/usr/local/lib/python3.6/dist-packages/torch/serialization.py in save(obj, f, pickle_module, pickle_protocol)
217 >>> torch.save(x, buffer)
218 """
--> 219 return _with_file_like(f, "wb", lambda f: _save(obj, f, pickle_module, pickle_protocol))
220
221
/usr/local/lib/python3.6/dist-packages/torch/serialization.py in _with_file_like(f, mode, body)
140 (sys.version_info[0] == 3 and isinstance(f, pathlib.Path)):
141 new_fd = True
--> 142 f = open(f, mode)
143 try:
144 return body(f)
FileNotFoundError: [Errno 2] No such file or directory: 'content/gdrive/My Drive/Machine Learning Models/kaggle_jigsaw_gru_iter_0.pth'
我已在我的驱动器中手动创建了该文件夹,只需要创建该文件。但错误仍然存在。不过,我确信不需要手动创建文件夹。问题是别的。我哪里错了?