使用文件名中的日期时间保存 Pytorch 模型 state_dict

Question

我一直在尝试使用
torch.save(agent.qnetwork_local.state_dict(), filename) 其中
filename = datetime.now().strftime('%d-%m-%y-%H:%M_dqnweights.pth')

保存 Pytorch 模型的 state_dict

type(filename) 返回str，torch.save() 应该没有问题，它应该输出一个非空文件。相反，我得到一个只有日期和时间的空文件，之后什么都没有。将日期和文件名放在中间会导致一个空文件，其中包含截止日期和时间之后的所有内容。

torch.save(agent.qnetwork_local.state_dict(), 'checkpoint1.pth') 并且任何时候我硬编码字符串都可以工作并给我预期的非空文件。

发生了什么，我该如何解决？

我在 Windows 10 上使用 Pytorch v1.8.1+cpu 在 Python v3.6.8 virtualenv 中运行此代码。

Answer 1

冒号是filename = datetime.now().strftime('%d-%m-%y-%H:%m_dqnweights.pth') 中的问题，因为它是在 Windows 上运行的。

将其更改为 filename = datetime.now().strftime('%d-%m-%y-%H_%M_dqnweights.pth') 可以正常工作。