diff --git a/torchgeo/trainers/classification.py b/torchgeo/trainers/classification.py index a1f934b3173..675595c767b 100644 --- a/torchgeo/trainers/classification.py +++ b/torchgeo/trainers/classification.py @@ -38,13 +38,13 @@ class ClassificationTask(pl.LightningModule): Supports any available `Timm model `_ - as an architecture choice. To see a list of available pretrained + as an architecture choice. To see a list of available models, you can do: .. code-block:: python import timm - print(timm.list_models(pretrained=True)) + print(timm.list_models()) """ def config_model(self) -> None: