hlca_map_classify.ipynb missing early_stopping_kwarg
c-westhoven opened this issue · 0 comments
c-westhoven commented
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},
}