Merge pull request #2 from nickp60/main-1

add message about convergence
nickp60 · Aug 8, 2023 · 6b39aa8 · 6b39aa8
2 parents aa1593b + 9f09be8
commit 6b39aa8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spectra/spectra.py b/spectra/spectra.py
@@ -576,7 +576,7 @@ def train(self,X, labels = None, lr_schedule = [1.0,.5,.1,.01,.001,.0001],num_ep
                     if verbose:
                         print("UPDATING LR TO " + str(lr_schedule[int(counter/3)]))
             last = loss.item() 
-
+        print("Reached converence at epoch " + i)
 
         #add all model parameters as attributes