From: François Fleuret <francois@fleuret.org>
Date: Tue, 28 Mar 2023 20:36:19 +0000 (+0200)
Subject: Update
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=beaver.git;a=commitdiff_plain;h=2ee976b3249254d1eb796678c5408ea45293489b

Update
---

diff --git a/beaver.py b/beaver.py
index f5b3563..7800527 100755
--- a/beaver.py
+++ b/beaver.py
@@ -265,6 +265,8 @@ def oneshot(gpt, learning_rate_scheduler, task):
 
     for n_epoch in range(args.nb_epochs):
         learning_rate = learning_rate_scheduler.get_learning_rate()
+        log_string(f"learning_rate {n_epoch} {learning_rate}")
+
         optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
 
         acc_train_loss, nb_train_samples = 0, 0
@@ -711,8 +713,7 @@ learning_rate_scheduler.reset()
 
 for n_epoch in range(nb_epochs_finished, args.nb_epochs):
     learning_rate = learning_rate_scheduler.get_learning_rate()
-
-    log_string(f"learning_rate {learning_rate}")
+    log_string(f"learning_rate {n_epoch} {learning_rate}")
 
     if args.optim == "sgd":
         optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)