Skip to content

Commit 8aae3d8

Browse files
kamalesh0406Cadene
authored andcommitted
Update README.md
1 parent 1254428 commit 8aae3d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ model = pretrainedmodels.__dict__[model_name](num_classes=1000, pretrained='imag
142142
model.eval()
143143
```
144144

145-
**Note**: By default, models will be downloaded to your `$HOME/.torch` folder. You can modify this behavior using the `$TORCH_MODEL_ZOO` variable as follow: `export TORCH_MODEL_ZOO="/local/pretrainedmodels`
145+
**Note**: By default, models will be downloaded to your `$HOME/.torch` folder. You can modify this behavior using the `$TORCH_HOME` variable as follow: `export TORCH_HOME="/local/pretrainedmodels"`
146146

147147
- To load an image and do a complete forward pass:
148148

0 commit comments

Comments
 (0)