When I use features = torch.FloatTensor(1,512).zero_.cuda()
in the python script, there is an error, why?
features = torch.zeros(1,512).cuda()
is ok!
When I use features = torch.FloatTensor(1,512).zero_.cuda()
in the python script, there is an error, why?
features = torch.zeros(1,512).cuda()
is ok!
Hi,
.zero_
is a method not attribute. You need to call it using parentheses:
features = torch.FloatTensor(1,512).zero_().cuda()
Bests