diff --git a/scvi/model/_totalvi.py b/scvi/model/_totalvi.py
index acf53ee28e..a0df889695 100644
--- 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,