小编Edu*_*era的帖子

我有一个 pytorch 图像分类器训练,我想暂停训练并在程序暂停时保存权重。我可以这样做吗?

我正在训练一个已经训练了几天的分类器,但我的问题是我没有在整个训练过程中编写代码来保存 .pt 检查点,所以我最终只会得到一个权重文件当程序完成其所有时期时。有没有办法立即暂停训练(PAUSE BREAK)并保存模型的权重?

python pytorch

5
推荐指数
1
解决办法
1103
查看次数

是否有 tensor.item() 等效于包含 pytorch 中的列表的张量?

在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)

我似乎无法在支持此类操作的文档中找到任何功能。

python machine-learning pytorch tensor

2
推荐指数
1
解决办法
2224
查看次数

标签 统计

python ×2

pytorch ×2

machine-learning ×1

tensor ×1