From c371a0336d5562e1dd9266f42ccf8434091100a4 Mon Sep 17 00:00:00 2001
From: Wenjie Wu <wuwenjie0102@gmail.com>
Date: Wed, 14 Aug 2019 19:23:37 +0800
Subject: [PATCH] Fix doc error for tensorflow CIFAR10

---
 tensorflow/CIFAR10/resnet/README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tensorflow/CIFAR10/resnet/README.md b/tensorflow/CIFAR10/resnet/README.md
index b8027c1..fe96eb6 100644
--- a/tensorflow/CIFAR10/resnet/README.md
+++ b/tensorflow/CIFAR10/resnet/README.md
@@ -70,7 +70,8 @@ cifar_input.py  README.md  resnet_main.py  resnet_model.py
 $ python3 resnet/resnet_main.py --train_data_path=cifar10/data_batch* \
                                 --log_root=/tmp/resnet_model \
                                 --train_dir=/tmp/resnet_model/train \
-                                --dataset='cifar10' \
+                                --dataset=cifar10 \
+                                --model=resnet20 \ 
                                 --num_gpus=1
 
 # While the model is training, you can also check on its progress using tensorboard: