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



我一直在尝试用
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')可以按预期工作。

相关内容

  • 没有找到相关文章

最新更新