From 1b2d0154a2ae21513ce96fa4bf91416cf1ed643f Mon Sep 17 00:00:00 2001 From: Nils Lehmann Date: Mon, 28 Nov 2022 08:38:42 +0100 Subject: [PATCH] docstring list_models --- torchgeo/trainers/classification.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: