What was the basis for parameter choices in tutorials?

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

1 Like

I also think would be good to know.