RAM model: RuntimeError: Expected object of scalar type Float but got scalar type Long

[Nivek92]

When I run

python3 train.py --model_name ram --dataset restaurant --learning_rate 1e-3 --num_epoch 200

I get

RuntimeError: Expected object of scalar type Float but got scalar type Long for sequence element 1 in sequence argument at position #1 'tensors'

[nghidinhit]

In locationed_memory() function, you should add the following line: u = u.type(torch.FloatTensor).to(self.opt.device)