a.h.23
1
Minimal Sample:
import torch
from typing import List
@torch.jit.script
def get_boxes(batch_bboxes=None):
s = torch.jit.annotate(List[List[torch.Tensor]], [])
s.append([torch.tensor(3.14)])
s.append([torch.tensor(42.23)])
bb = torch.tensor(s)
print(bb)
get_boxes()
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
bb = torch.tensor(b)
RuntimeError: Input must be of ints, floats, or bools, got Tensor
Hi @a.h.23,
Have you tried using torch.stack
to convert the List of Tensors to Tensor, then repeat again for the 2nd list?
a.h.23
3
Thx AlphaBetaGamma96 !
My solution was now this:
ccc= torch.empty((0,1))
for bbox in pb:
ch = torch.stack([category], dim=0)
ch = torch.unsqueeze(ch, 0)
ccc = torch.cat((ccc, ch))
I also opend a Issue at the pytorch repo: