From a0c6d8fd853cb504d7be89028721120f244121f2 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 15:34:47 +0200
Subject: [PATCH] datasets
---
main_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main_custom.py b/main_custom.py
index d04876c..df172b3 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -43,7 +43,7 @@ def train(model, data_train, epoch, optimizer, criterion_rt, criterion_intensity
optimizer.zero_grad()
loss.backward()
optimizer.step()
- print(i,'/',len(data_train))
+ print(i,'/',len(data_train.dataset))
if wandb is not None:
wdb.log({"train rt loss": losses_rt / len(data_train), "train int loss": losses_int / len(data_train),
--
GitLab