Skip to content

GitLab

"...text-generation-inference.git" did not exist on "52e48739a57e29ba47c238b2bbf06a391066da57"
Unverified Commit 107dec64 authored by Stephen Matthews's avatar Stephen Matthews Committed by GitHub
Browse files

Fix TORCH_HOME environment variable in docs (#4843) 



* Fix TORCH_HOME environment variable in docs

* Update docs with torch.hub

Updated the reference for loading a model to `torch.hub`.
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent b1fc2903
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docs/source/models.rst
View file @ 107dec64
...@@ -127,8 +127,8 @@ These can be constructed by passing ``pretrained=True``: ...@@ -127,8 +127,8 @@ These can be constructed by passing ``pretrained=True``:
regnet_x_32gf = models.regnet_x_32gf(pretrained=True) regnet_x_32gf = models.regnet_x_32gf(pretrained=True)
Instancing a pre-trained model will download its weights to a cache directory. 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 This directory can be set using the `TORCH_HOME` environment variable. See
:func:`torch.utils.model_zoo.load_url` for details. :func:`torch.hub.load_state_dict_from_url` for details.
Some models use modules which have different training and evaluation Some models use modules which have different training and evaluation
behavior, such as batch normalization. To switch between these modes, use behavior, such as batch normalization. To switch between these modes, use
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment