From 9d3b814b37138d5c6d3608506797440518337f95 Mon Sep 17 00:00:00 2001 From: Erick Matsen Date: Mon, 20 May 2024 02:20:41 -0700 Subject: [PATCH] format --- netam/framework.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/netam/framework.py b/netam/framework.py index 28efd967..e4ceb0dd 100644 --- a/netam/framework.py +++ b/netam/framework.py @@ -649,8 +649,11 @@ def joint_train(self, epochs=100, cycle_count=2, training_method="full"): If training_method is "yun", then we use Yun's approximation to the branch lengths. If training_method is "fixed", then we fix the branch lengths and only optimize the model. - We reset the optimization after each cycle, and we use a learning rate schedule that - uses a weighted geometric mean of the current learning rate and the initial learning rate that progressively moves towards keeping the current learning rate as the cycles progress. + We reset the optimization after each cycle, and we use a learning rate + schedule that uses a weighted geometric mean of the current learning + rate and the initial learning rate that progressively moves towards + keeping the current learning rate as the cycles progress. + """ if training_method == "full": optimize_branch_lengths = self.standardize_and_optimize_branch_lengths