Кто-нибудь может сказать мне, как проверить, существует ли модель Pytorch, и если существует, удалить ее и заменить новой?

#python #pytorch

Вопрос:

Поэтому я сохраняю множество моделей torch для обучения с разным размером пакета и эпохами, и модели сохраняются со строками epoch и batchsize. В принципе, я иногда меняю гиперпараметры некоторых слоев и некоторые дополнения, чтобы проверить результаты прогнозирования, но если модель torch существует, я хочу удалить ее и заменить новой.

1. Просто используйте torch.save с тем же путем / именем. Это переопределит сохранение старой модели.

2. Я думал, что она будет использовать предыдущие оценки и обновлять их. В любом случае, спасибо. 🙂

3. Нет, она просто записывает все ваши текущие параметры в файл.

Ответ №1:

Самое простое решение — просто сохранить модель с тем же именем, по существу перезаписав существующую. Это эквивалентно проверке, существует ли она, удалению и последующему сохранению.

Если вы хотите явно проверить, существует ли она, вы можете легко сделать это с помощью os.

 import os
if os.path.exists('path/to/model.pth'):  # checking if there is a file with this name
    os.remove('path/to/model.pth')  # deleting the file
    torch.save(model, 'path/to/model.pth')  # saving a new model with the same name