From ba503db4f9c0f9949d03a43712c50d293539d6c9 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 10 Mar 2025 14:46:55 +0100
Subject: [PATCH] model cuda loading
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index 06be979..cd8556e 100644
--- a/main.py
+++ b/main.py
@@ -34,7 +34,7 @@ def train(model, data_train, optimizer, loss_function, epoch):
optimizer.step()
losses = losses/len(data_train.dataset)
acc = acc/len(data_train.dataset)
- print('Test epoch {}, loss : {:.3f} acc : {:.3f}'.format(epoch,losses,acc))
+ print('Train epoch {}, loss : {:.3f} acc : {:.3f}'.format(epoch,losses,acc))
return losses, acc
def test(model, data_test, loss_function, epoch):
--
GitLab