Как скачать возврат с гугл коллаба?

Question

Я использую Google Collab для обучения моей модели LSTM. Я хочу загрузить файл модели на свой ноутбук. У меня есть такой код:

model.fit(x_train, y_train, epochs=1, batch_size=1, verbose=2)
save_file_name = pair + ".h5"
download(model.save(save_file_name))

Я пытался использовать download (), но у меня ничего не получается. Есть ли способ скачать?

Answer 1

Я предполагаю, что вы используете Keras, потому что метод model.save() возвращает None, поэтому, если вы укажете его как параметр метода download(), произойдет ошибка. Сначала сохраните модель, а затем загрузите ее:

model.save('model_name.h5')
download('model_name.h5')

И убедитесь, что вы находитесь в правильном каталоге, содержащем файл model_name.h5. Вы можете проверить это с помощью команды !ls.