From 6f48c9f91c0d2b897006a66726489eae3895c637 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 25 Oct 2024 13:41:24 +0200
Subject: [PATCH] datasets
---
main_ray_tune.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_ray_tune.py b/main_ray_tune.py
index 8554204..00d12bc 100644
--- a/main_ray_tune.py
+++ b/main_ray_tune.py
@@ -158,7 +158,7 @@ def train_model(config, args):
loss_rt = criterion_rt(rt, pred_rt)
loss_int = criterion_intensity(intensity, pred_int)
loss = loss_rt + loss_int
- val_loss += loss.item().numpy()
+ val_loss += loss.item()
val_steps += 1
# Here we save a checkpoint. It is automatically registered with
--
GitLab