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
