火炬，.pth文件保存在哪里？

Question

这是一个很痛苦的问题，但我想不出来。因此，我将模型权值保存为文件，显然它们保存在我的计算机上的某个地方，我可以再次加载它们，并在.pth文件之间切换。但我完全不知道他们被救到哪里去了。

FILE ="test.pth"
torch.save(policy.state_dict(), FILE)

policy.load_state_dict(torch.load("test.pth"))
policy.eval()

任何帮助都非常感谢，真的。谢谢。

Answer 1

这些文件保存在您执行代码的目录中。您可以找到以下目录：

import os
print(os.getcwd()) # get current working directory