From a9276df01fc262a241de7c39d9b8823ad40158d1 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 24 Sep 2024 12:18:56 +0200
Subject: [PATCH] dataset rain ISA
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index e34b7d1..fb99a74 100644
--- a/main_custom.py
+++ b/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():
--
GitLab