Backport PR #1639: Fix toalvi max_epochs (#1643)

Co-authored-by: Valeh Valiollah Pour Amiri <[email protected]>
scverse · Aug 20, 2022 · 78e68bb · 78e68bb
1 parent 5ea15db
commit 78e68bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/model/_totalvi.py b/scvi/model/_totalvi.py
@@ -192,7 +192,7 @@ def __init__(
 
     def train(
         self,
-        max_epochs: Optional[int] = 400,
+        max_epochs: Optional[int] = None,
         lr: float = 4e-3,
         use_gpu: Optional[Union[str, int, bool]] = None,
         train_size: float = 0.9,