There is ERROR while running ' 3.Train SPICE-Self' in train.md.

[DonghoonPark12]

File "/home/user/XXX/SPICE/tools/train_self_v2.py", line 236, in main_worker
train(train_loader, model, optimizer, epoch, cfg)
File "/home/user/XXX/SPICE/tools/train_self_v2.py", line 472, in train
images_trans.append(images_trans_l[idx_select[h], :, :, :])
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

How can I solve it??