dataset rain ISA

main_custom.py

@@ -25,7 +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'])
+        print(data_train.dataset.data['Sequence'])
         for seq, charge, rt, intensity in data_train:
             rt, intensity = rt.float(), intensity.float()
             if torch.cuda.is_available():