fix max_epochs in totalvi train

scverse · Aug 16, 2022 · 20df1b1 · 20df1b1
1 parent c6b2b12
commit 20df1b1
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,