Google Colab: «Невозможно подключиться к среде выполнения» после загрузки модели Pytorch из локальной системы.

Question

Я использую простой (не обязательно эффективный) метод для сохранения модели Pytorch.

import torch
from google.colab import files

torch.save(model, filename) # save a trained model on the VM
files.download(filename) # download the model to local

best_model = files.upload() # select the model just downloaded
best_model[filename] # access the model

Colab отключается во время выполнения последней строки, и при нажатии на вкладку RECONNECT всегда отображается ALLOCATING ->CONNECTING (ошибка, с сообщением «невозможно подключиться к среде выполнения» в левом нижнем углу) -> RECONNECT.В то же время выполнение любой из ячеек выдает сообщение об ошибке «Не удалось выполнить ячейку. Не удалось отправить сообщение об исполнении во время выполнения: [object CloseEvent]»

Я знаю, что это связано с последней строкой, посколькуЯ могу успешно подключиться к другим своим аккаунтам Google, которые этого не выполняют.

Почему это происходит?Похоже, аккаунты Google, которые выполнили последнюю строку, больше не могут подключаться к среде выполнения.

Редактировать:

Однажды ночью я могу снова подключиться к аккаунту Googleпосле истечения сеанса.Я просто попытался использовать подход в комментарии и обнаружил, что только files.upload() модель Pytorch приведет к проблеме.После завершения загрузки Colab отключается.

Answer 1

Попробуйте отключить блокировку рекламы.Работал на меня

Answer 2

(я написал этот ответ перед прочтением вашего обновления. Думаю, это может помочь.)

files.upload() только для загрузки файлов.У нас нет оснований ожидать, что он вернет немного pytorch type/model.

Когда вы вызываете a = files.upload(), a является словарем имени файла - массива больших байтов.

{'my_image.png': b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR....' }
type(a['my_image.png'])

Так же, как когда вы делаете open('my_image', 'b').read()

Итак, я думаю, что следующая строка best_model[filename] попытается напечатать весь массив огромных байтов, который содержит ошибки в колабе.