`torch_mist.estimators.factories`

Module Contents

Functions

instantiate_estimator(...)

Attributes

DEFAULT_HIDDEN_DIMS

torch_mist.estimators.factories.DEFAULT_HIDDEN_DIMS = [128]

torch_mist.estimators.factories.instantiate_estimator(estimator_name: str, x_dim: int | None = None, y_dim: int | None = None, **kwargs) → torch_mist.estimators.base.MIEstimator