-
-
Notifications
You must be signed in to change notification settings - Fork 17.2k
Closed
Description
Why does model(img)
works but model(torch.Tensor(img))
does not, despite having the same shape resulting in the following error?
RuntimeError: Sizes of tensors must match except in dimension 2. Got 1 and 2 (The offending index is 0)
img
is of shape (720, 1280, 3)
and of type int
.
Thanks
JohnTailor
Metadata
Metadata
Assignees
Labels
No labels