scarches icon indicating copy to clipboard operation
scarches copied to clipboard

hlca_map_classify.ipynb missing early_stopping_kwarg

Open c-westhoven opened this issue 1 year ago • 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},
}

c-westhoven avatar Oct 11 '23 13:10 c-westhoven