From 936fc49c68cc27289e61453d09480c822eb1fd81 Mon Sep 17 00:00:00 2001 From: Nils Lehmann Date: Mon, 28 Nov 2022 08:40:28 +0100 Subject: [PATCH] docstring list_models --- torchgeo/trainers/regression.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchgeo/trainers/regression.py b/torchgeo/trainers/regression.py index a475baf4de7..e9667486deb 100644 --- a/torchgeo/trainers/regression.py +++ b/torchgeo/trainers/regression.py @@ -30,13 +30,13 @@ class RegressionTask(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_task(self) -> None: