我正在训练一个已经训练了几天的分类器,但我的问题是我没有在整个训练过程中编写代码来保存 .pt 检查点,所以我最终只会得到一个权重文件当程序完成其所有时期时。有没有办法立即暂停训练(PAUSE BREAK)并保存模型的权重?
在pytorch中,如果我定义一个单元素张量如下:
>>> import torch
>>> target1 = torch.tensor([5])
Run Code Online (Sandbox Code Playgroud)
我可以像这样提取一个元素的值:
>>> target1.item()
5
Run Code Online (Sandbox Code Playgroud)
我想知道的是,当我的张量定义为:
target2 = torch.tensor([[5], [5], [5], [5]])
Run Code Online (Sandbox Code Playgroud)
是否有某种方法(类似于或不类似于上面的.item())将其所有条目提取到一个列表中,例如:
>>> target2.(something)
[5, 5, 5, 5]
Run Code Online (Sandbox Code Playgroud)
我似乎无法在支持此类操作的文档中找到任何功能。