From b04dcb9686bbd4120febf1395cb0a5894ff1cb3c Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 22 Oct 2024 17:06:27 +0200
Subject: [PATCH] datasets
---
main_custom.py | 2 --
1 file changed, 2 deletions(-)
diff --git a/main_custom.py b/main_custom.py
index a0f6489..a8db447 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -116,8 +116,6 @@ def eval(model, data_val, epoch, criterion_rt, criterion_intensity, metric_rt, m
if torch.cuda.is_available():
seq, charge, rt, intensity = seq.cuda(), charge.cuda(), rt.cuda(), intensity.cuda()
pred_rt, pred_int = model.forward(seq, charge)
- print(rt)
- print(rt.shape,pred_rt.shape)
loss_rt = criterion_rt(rt, pred_rt)
loss_int = criterion_intensity(intensity, pred_int)
losses_rt += loss_rt.item()
--
GitLab