From 80c232514370d6e9b862eda5f4498588c51fae32 Mon Sep 17 00:00:00 2001
From: Huy Phan <huyvnphan@gmail.com>
Date: Sun, 10 Jan 2021 21:20:43 -0500
Subject: [PATCH] Update README.md

---
 README.md | 1 +
 1 file changed, 1 insertion(+)

diff --git a/README.md b/README.md
index 5b30da5..4dfa9b2 100644
--- 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