我一直在尝试用torch.save(agent.qnetwork_local.state_dict(), filename)
保存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虚拟机中运行这段代码。
冒号是filename = datetime.now().strftime('%d-%m-%y-%H:%m_dqnweights.pth')
中的问题,因为它在windows上运行。
将其更改为filename = datetime.now().strftime('%d-%m-%y-%H_%M_dqnweights.pth')
可以按预期工作。