Commit b94e2c82 authored by Samuel Fadel's avatar Samuel Fadel Committed by Soumith Chintala
Browse files

Mention TORCH_MODEL_ZOO in models documentation. (#624)

parent 86001a87
...@@ -36,6 +36,10 @@ These can be constructed by passing ``pretrained=True``: ...@@ -36,6 +36,10 @@ These can be constructed by passing ``pretrained=True``:
densenet = models.densenet161(pretrained=True) densenet = models.densenet161(pretrained=True)
inception = models.inception_v3(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 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
``model.train()`` or ``model.eval()`` as appropriate. See ``model.train()`` or ``model.eval()`` as appropriate. See
......
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