我一直在试图保存一个Pytorch模型的state_dict
torch.save(agent.qnetwork_local.state_dict(), filename)
,其中
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')
和任何时候我硬编码字符串工作,并给我期望的非空文件。
怎么回事,我该怎么解决这个问题?
我是在Pythonv3.6.8Virtualenv中运行这段代码的,它在Windows10上使用了Pytorch v1.8.1+cpu。
发布于 2021-06-16 07:10:49
冒号是filename = datetime.now().strftime('%d-%m-%y-%H:%m_dqnweights.pth')
中的问题,因为它在windows上运行。
将其更改为filename = datetime.now().strftime('%d-%m-%y-%H_%M_dqnweights.pth')
会像预期的那样工作。
https://stackoverflow.com/questions/67958129
复制