such as

[vlior]

File "main_v1.0.py", line 90, in
model = train(model, train_loader, eval_loader, args)
File "/home/bigdata2/xnq/DMRM-master/train.py", line 67, in train
train_data = next(train_iter)
IndexError: arrays used as indices must be of integer (or boolean) type

[phellonchen]

I guess you could check the pytorch version. The code is tested on v0.3.0.