Update README.md

huyvnphan · Jan 11, 2021 · 80c2325 · 80c2325
1 parent 038caf6
commit 80c2325
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -41,6 +41,7 @@ my_model = vgg11_bn()
 
 # Pretrained model
 my_model = vgg11_bn(pretrained=True)
+my_model.eval() # for evaluation
 ```
 
 If you use your own images, all models expect data to be in range [0, 1] then normalized by