docstring list_models

microsoft · Nov 28, 2022 · 1b2d015 · 1b2d015
1 parent d58beb1
commit 1b2d015
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchgeo/trainers/classification.py b/torchgeo/trainers/classification.py
@@ -38,13 +38,13 @@ class ClassificationTask(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_model(self) -> None: