From 9f09be86f101106c6eea934b9c64349a894e3647 Mon Sep 17 00:00:00 2001 From: Nick Waters Date: Tue, 8 Aug 2023 09:09:42 -0400 Subject: [PATCH 1/2] add message about convergence --- spectra/spectra.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 From cea7a89a4511f035496d8c3c741d0370c2992cb4 Mon Sep 17 00:00:00 2001 From: Nick Waters Date: Tue, 8 Aug 2023 09:12:51 -0400 Subject: [PATCH 2/2] add message to gpu spectra --- spectra/spectra_gpu.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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: