Merge pull request #70 from calico/testbranch

Allow specification of validation loss check interval

solo/solo.py

@@ -193,6 +193,7 @@ def main():
     batch_key = params.get("batch_key", None)
     batch_size = params.get("batch_size", 128)
     valid_pct = params.get("valid_pct", 0.1)
+    check_val_every_n_epoch = params.get("check_val_every_n_epoch", 5)
     learning_rate = params.get("learning_rate", 1e-3)
     stopping_params = {"patience": params.get("patience", 8), "min_delta": 0}
 
@@ -231,11 +232,10 @@ def main():
             "lr_min": 1e-4,
             "lr_scheduler_metric": "reconstruction_loss_validation",
         }
-
         vae.train(
             max_epochs=2000,
             validation_size=valid_pct,
-            check_val_every_n_epoch=5,
+            check_val_every_n_epoch=check_val_every_n_epoch,
             plan_kwargs=plan_kwargs,
             callbacks=scvi_callbacks,
         )