Nguyen Dinh Nghi
Results
2
comments of
Nguyen Dinh Nghi
My pytorch version is: 1.0.1 Any suggestion? Please!
In locationed_memory() function, you should add the following line: u = u.type(torch.FloatTensor).to(self.opt.device)