datasets

main_custom.py

@@ -116,6 +116,7 @@ def eval(model, data_val, epoch, criterion_rt, criterion_intensity, metric_rt, m
             if torch.cuda.is_available():
                 seq, charge, rt, intensity = seq.cuda(), charge.cuda(), rt.cuda(), intensity.cuda()
             pred_rt, pred_int = model.forward(seq, charge)
+            print(rt.shape,pred_rt.shape)
             loss_rt = criterion_rt(rt, pred_rt)
             loss_int = criterion_intensity(intensity, pred_int)
             losses_rt += loss_rt.item()