diff --git a/spectra/spectra.py b/spectra/spectra.py
index 69fa446..34a3777 100644
--- 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 
 
diff --git a/spectra/spectra_gpu.py b/spectra/spectra_gpu.py
index 54b5f44..ef0f22e 100644
--- a/spectra/spectra_gpu.py
+++ b/spectra/spectra_gpu.py
@@ -545,7 +545,8 @@ def train(self,X, labels = None, lr_schedule = [.5,.1,.01,.001,.0001], num_epoch
                             print("UPDATING LR TO " + str(lr_schedule[int(counter/skip)]))
                 last = loss.item() 
 
-
+        print("Reached converence at epoch " + i)
+        
         #add all model parameters as attributes 
 
         if self.use_cell_types: