Closed rracinskij closed 7 years ago
It means you're returning a cudatensor from the lua side, into the python side.
Your easiest option would be to convert it to a FloatTensor on the lua side, before passing it into the python.
(so eg something like return output:float()
)
Yes, it works. The error was referenced to
local output = self.net:forward(torch.rand(1,3,112,112):cuda())
and not to the return output
, so it was a bit confusing.
Many thanks!
Cool :-)
Hi, I get a strange error while trying a forward pass through a loaded Torch model: