torch.save (модель, ПУТЬ) ошибка: NotADirectoryError: [Errno 20] Не каталог: '~ / KoGPT2_checkpoint.tar'

Question

Я использую pytorch, и я хочу использовать контрольную точку pytorch

это мой код

import os

save_path = 'drive/My Drive/Colab Notebooks/KoGPT2_checkpoint/'
torch.save(model, os.path.join(save_path+'KoGPT2_checkpoint.tar')) 

, и в сообщении об ошибке говорится:

/usr/local/lib/python3.6/dist-packages/torch/serialization.py in __init__(self, name, mode)
    213 class _open_file(_opener):
    214     def __init__(self, name, mode):
--> 215         super(_open_file, self).__init__(open(name, mode))
    216 
    217     def __exit__(self, *args):

NotADirectoryError: [Errno 20] Not a directory: 'drive/My Drive/Colab Notebooks/KoGPT2_checkpoint/KoGPT2_checkpoint.tar'

Я также протестировал такой код, чтобы решить эту

import os

save_path = 'drive/My Drive/Colab Notebooks/KoGPT2_checkpoint'
torch.save(model, 'drive/My Drive/Colab Notebooks/KoGPT2_checkpoint') 

, но проблема в том, что я не мог создать файл .tar в моем colab. Я мог создать только файл «KoGPT2_checkpoint», у которого не было никакого расширения

Я не знаю, почему я получаю эту ошибку Не могли бы вы, ребята, рассказать мне, как создавать контрольные точки с помощью pytorch?

Answer 1

Указанный путь не существует.

Возможно, вы не подключили диск с помощью Google Colab. Попробуйте смонтировать и проверьте путь.

from google.colab import drive
drive.mount('/gdrive')

Затем укажите соответствующий путь

save_path = '/gdrive/My Drive/Colab Notebooks/KoGPT2_checkpoint'