We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d1e9a86 commit 9383f28Copy full SHA for 9383f28
timm/models/_factory.py
@@ -89,6 +89,10 @@ def create_model(
89
>>> model = create_model('mobilenetv3_large_100', pretrained=True, num_classes=10)
90
>>> model.num_classes
91
10
92
+
93
+ >>> # Create a Dinov2 small model with pretrained weights and save weights in a custom directory.
94
+ >>> model = create_model('vit_small_patch14_dinov2.lvd142m', pretrained=True, cache_dir="/data/my-models")
95
+ >>> # Data will be stored at `/data/my-models/models--timm--vit_small_patch14_dinov2.lvd142m/`
96
```
97
"""
98
# Parameters that aren't supported by all models or are intended to only override model defaults if set
0 commit comments