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

Question

Таким образом, я сохраняю множество моделей факелов для обучения с разными размерами пакетов и эпохами, а модели сохраняются со строками эпох и пакетов. Обычно я иногда изменяю некоторые гиперпараметры слоев и некоторые дополнения, чтобы проверить результаты прогнозирования, но если модель факела есть, я хочу удалить ее и заменить новой.

Answer 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