Hello,
I was wondering what the basis was for the parameters chosen in the tutorials such as “Reference mapping with scvi-tools” and “Querying the Human Lung Cell Atlas”. Specifically, the number of epochs and the training arguments:
arches_params = dict(
use_layer_norm="both",
use_batch_norm="none",
encode_covariates=True,
dropout_rate=0.2,
n_layers=2,
)
surgery_epochs = 500
train_kwargs_surgery = {
"early_stopping": True,
"early_stopping_monitor": "elbo_train",
"early_stopping_patience": 10,
"early_stopping_min_delta": 0.001,
"plan_kwargs": {"weight_decay": 0.0},
}
Were these parameters chosen from some type of search or is there some rule of thumb that was applied?
Cheers,
Tony