在文件名中保存带有日期时间的 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')
按预期工作。
我一直在尝试用
保存 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')
按预期工作。