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

Question

torch.save(agent.qnetwork_local.state_dict(), filename)我一直在尝试用where保存 Pytorch 模型的 state_dict
filename = datetime.now().strftime('%d-%m-%y-%H:%M_dqnweights.pth')

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')按预期工作。