Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
b94e2c82
Commit
b94e2c82
authored
Oct 11, 2018
by
Samuel Fadel
Committed by
Soumith Chintala
Oct 11, 2018
Browse files
Mention TORCH_MODEL_ZOO in models documentation. (#624)
parent
86001a87
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
docs/source/models.rst
docs/source/models.rst
+4
-0
No files found.
docs/source/models.rst
View file @
b94e2c82
...
...
@@ -36,6 +36,10 @@ These can be constructed by passing ``pretrained=True``:
densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(pretrained=True)
Instancing a pre-trained model will download its weights to a cache directory.
This directory can be set using the `TORCH_MODEL_ZOO` environment variable. See
:func:`torch.utils.model_zoo.load_url` for details.
Some models use modules which have different training and evaluation
behavior, such as batch normalization. To switch between these modes, use
``model.train()`` or ``model.eval()`` as appropriate. See
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment