From 5217ca19b47abb4f0e413b7e7f68d1d82548d899 Mon Sep 17 00:00:00 2001
From: Stefan Kesselheim <s.kesselheim@fz-juelich.de>
Date: Fri, 27 Sep 2024 08:10:49 +0200
Subject: [PATCH] update

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index e5f06ee..cc4b9d6 100644
--- a/train.py
+++ b/train.py
@@ -83,7 +83,7 @@ model=model.cuda()
 
 loss_fn = torch.nn.CrossEntropyLoss()
 
-optimizer=torch.optim.Adam(model.parameters(), lr=0.001)
+optimizer=torch.optim.Adam(model.parameters(), lr=0.0001)
 
 num_epochs=20
 args=types.SimpleNamespace(dry_run=False, log_interval=16)
-- 
GitLab