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