Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

(nit) Fix <Tip> style in docs #2336

Closed
wants to merge 1 commit into from
Closed

(nit) Fix <Tip> style in docs #2336

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions hfdocs/source/quickstart.mdx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,9 @@ Here, we load the pretrained `mobilenetv3_large_100` model.
```

<Tip>
Note: The returned PyTorch model is set to train mode by default, so you must call .eval() on it if you plan to use it for inference.

**Note:** The returned PyTorch model is set to train mode by default, so you must call `.eval()` on it if you plan to use it for inference.

</Tip>

## List Models with Pretrained Weights
Expand Down Expand Up @@ -151,7 +153,9 @@ Compose(
```

<Tip>
Note: Here, the pretrained model's config happens to be the same as the generic config we made earlier. This is not always the case. So, it's safer to use the data config to create the transform as we did here instead of using the generic transform.

**Note:** Here, the pretrained model's config happens to be the same as the generic config we made earlier. This is not always the case. So, it's safer to use the data config to create the transform as we did here instead of using the generic transform.

</Tip>

## Using Pretrained Models for Inference
Expand Down
4 changes: 3 additions & 1 deletion timm/models/_factory.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,8 +51,10 @@ def create_model(
Lookup model's entrypoint function and pass relevant args to create a new model.

<Tip>
**kwargs will be passed through entrypoint fn to ``timm.models.build_model_with_cfg()``

**kwargs will be passed through entrypoint fn to `timm.models.build_model_with_cfg()`
and then the model class __init__(). kwargs values set to None are pruned before passing.

</Tip>

Args:
Expand Down
Loading