theislab/scarches

hlca_map_classify.ipynb missing early_stopping_kwarg

Opened this issue · 0 comments

Hi,
I noticed that in the hlca_map_classify tutorial, the early stopping is not actually activated, because the early_stopping is not set to true, and so the training goes through all 500 surgery epochs.

From tutorial:

surgery_epochs = 500
early_stopping_kwargs_surgery = {
    "early_stopping_monitor": "elbo_train",
    "early_stopping_patience": 10,
    "early_stopping_min_delta": 0.001,
    "plan_kwargs": {"weight_decay": 0.0},
}