docstring list_models

microsoft · Nov 28, 2022 · 936fc49 · 936fc49
1 parent ae5f9a8
commit 936fc49
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchgeo/trainers/regression.py b/torchgeo/trainers/regression.py
@@ -30,13 +30,13 @@ class RegressionTask(pl.LightningModule):
 
     Supports any available `Timm model
     <https://rwightman.github.io/pytorch-image-models/>`_
-    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: