Skip to content

Commit

Permalink
Merge pull request #2358 from turicas/cache_dir
Browse files Browse the repository at this point in the history
Add cache_dir example
  • Loading branch information
rwightman authored Dec 6, 2024
2 parents b09f81c + a1d219c commit 9cec2f1
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions timm/models/_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,6 +89,10 @@ def create_model(
>>> model = create_model('mobilenetv3_large_100', pretrained=True, num_classes=10)
>>> model.num_classes
10
>>> # Create a Dinov2 small model with pretrained weights and save weights in a custom directory.
>>> model = create_model('vit_small_patch14_dinov2.lvd142m', pretrained=True, cache_dir="/data/my-models")
>>> # Data will be stored at `/data/my-models/models--timm--vit_small_patch14_dinov2.lvd142m/`
```
"""
# Parameters that aren't supported by all models or are intended to only override model defaults if set
Expand Down

0 comments on commit 9cec2f1

Please sign in to comment.