diff --git a/main_custom.py b/main_custom.py
index cca8c9ce9084ed12cfb64c0f6378451458a581ba..e34b7d1155adcdf2fe8da5cddbd7bbf162e98c25 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -25,6 +25,7 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity
     for param in model.parameters():
         param.requires_grad = True
     if forward == 'both':
+        print(data_train.dataset['Sequence'])
         for seq, charge, rt, intensity in data_train:
             rt, intensity = rt.float(), intensity.float()
             if torch.cuda.is_available():