datasets

main_ray_tune.py

@@ -158,7 +158,7 @@ def train_model(config, args):
                     loss_rt = criterion_rt(rt, pred_rt)
                     loss_int = criterion_intensity(intensity, pred_int)
                     loss = loss_rt + loss_int
-                val_loss += loss.item().numpy()
+                val_loss += loss.item()
                 val_steps += 1
 
         # Here we save a checkpoint. It is automatically registered with