Question

因此，我保存了很多用于训练的火炬模型，并且具有不同的batchsize和epoch，并且使用epoch和batchsize的字符串保存了模型。基本上，我有时会更改一些层的超参数和一些扩充来检查预测结果，但是如果有割炬模型，我想将其删除并用新模型替换。

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

谁能告诉我如何检查Pytorch模型是否存在，如果存在，请将其删除并替换为新模型？

1 个答案: