fix : error device cuda

image_ref/main_ray.py

@@ -83,9 +83,6 @@ def train_model(config,args):
                 imana = imana.cuda()
                 img_ref = img_ref.cuda()
                 label = label.cuda()
-            if torch.cuda.device_count() > 1:
-                pred_logits = model.module.forward(imaer, imana, img_ref)
-            else:
                 pred_logits = model.forward(imaer, imana, img_ref)