Evaluation Error in IDOL

[wjf5203]

Evaluation Error in IDOL

[aylinaydincs]

In my server there are 20 cpu's and 2 gpu's

[aylinaydincs]

Still I couldn't solve the error "RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)"
It comes from inference() method. I think the trainable tensors are keeping different devices. Should I change the self.merge_on_cpu = cfg.MODEL.IDOL.MERGE_ON_CPU in idol.py?

[ldknight]

You can add the following two lines of code below the error location. I successfully run the inference on a single gpu.
################################################
if not isinstance(indices,list):
indices=indices.cpu()
################################################

[aylinaydincs]

Thank you for your suggestion, I think it works!