在文件名中保存带有日期时间的 Pytorch 模型 state_dict

Save Pytorch model state_dict with datetime in filename

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

type(filename) returns str 这应该不是 torch.save() 的问题,它应该输出一个非空文件。相反,我得到一个只有日期和时间的空文件,之后什么都没有。将日期和文件名放在文件名中间会生成一个空文件,其中包含日期和时间之后的所有内容。

torch.save(agent.qnetwork_local.state_dict(), 'checkpoint1.pth') 每当我对字符串进行硬编码时,它就会工作并为我提供预期的非空文件。

这是怎么回事,我该如何解决?

我 运行 这段代码在 Python v3.6.8 virtualenv 和 Pytorch v1.8.1+cpu Windows 10.

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

将其更改为 filename = datetime.now().strftime('%d-%m-%y-%H_%M_dqnweights.pth') 按预期工作。