有谁能告诉我如何检查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