Skip to content

Commit 61d4861

Browse files
committed
Update Chinese_OCR.py
Solved a tiny little problem 😄
1 parent 1038bf1 commit 61d4861

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ocr/Chinese_OCR.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
#! /usr/bin/env python
1+
#! /usr/bin/env python
22
# -*- coding: utf-8 -*-
33

44
# top 1 accuracy 0.99826 top 5 accuracy 0.99989
@@ -45,7 +45,7 @@
4545

4646
tf.app.flags.DEFINE_boolean('restore', False, 'whether to restore from checkpoint')
4747
tf.app.flags.DEFINE_boolean('epoch', 1, 'Number of epoches')
48-
tf.app.flags.DEFINE_boolean('batch_size', 128, 'Validation batch size')
48+
tf.app.flags.DEFINE_integer('batch_size', 128, 'Validation batch size')
4949
tf.app.flags.DEFINE_string('mode', 'validation', 'Running mode. One of {"train", "valid", "test"}')
5050

5151
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)

0 commit comments

Comments
 (0)