diff --git a/docs/source/models.rst b/docs/source/models.rst index 0aa9c47a061..8f918514fb7 100644 --- a/docs/source/models.rst +++ b/docs/source/models.rst @@ -127,8 +127,8 @@ These can be constructed by passing ``pretrained=True``: regnet_x_32gf = models.regnet_x_32gf(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. +This directory can be set using the `TORCH_HOME` environment variable. See +:func:`torch.hub.load_state_dict_from_url` for details. Some models use modules which have different training and evaluation behavior, such as batch normalization. To switch between these modes, use