datasets

main_custom.py

@@ -26,6 +26,7 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity
     if forward == 'both':
         i=0
         for seq, charge, rt, intensity in data_train:
+            i+=seq.shape[0]
             rt, intensity = rt.float(), intensity.float()
             if torch.cuda.is_available():
                 seq, charge, rt, intensity = seq.cuda(), charge.cuda(), rt.cuda(), intensity.cuda()