diff --git a/main_custom.py b/main_custom.py
index daf954fe6ec1cd9b8ef1d6d3dcc6e68ef6c14b5c..fec2d98427eae0a414849090dc94a461eb614c6b 100644
--- a/main_custom.py
+++ b/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)
             print(rt.shape,pred_rt.shape)
             loss_rt = criterion_rt(rt, pred_rt)
             loss_int = criterion_intensity(intensity, pred_int)
@@ -199,8 +200,8 @@ def run(epochs, eval_inter, save_inter, model, data_train, data_val, data_test,
 
     else :
         for e in range(1, epochs + 1):
-            train(model, data_train, e, optimizer, criterion_rt, criterion_intensity, metric_rt, metric_intensity, forward,
-                  wandb=wandb)
+            # train(model, data_train, e, optimizer, criterion_rt, criterion_intensity, metric_rt, metric_intensity, forward,
+            #       wandb=wandb)
             if e % eval_inter == 0:
                 eval(model, data_val, e, criterion_rt, criterion_intensity, metric_rt, metric_intensity, forward,
                      wandb=wandb)