Skip to content

Commit 70e55a1

Browse files
committed
spacing
1 parent 00e6186 commit 70e55a1

File tree

4 files changed

+36
-17
lines changed

4 files changed

+36
-17
lines changed

tutorials/image/cifar10/cifar10.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,14 @@
4949
parser = argparse.ArgumentParser()
5050

5151
# Basic model parameters.
52-
parser.add_argument('--batch_size', type=int, default=128, help='Number of images to process in a batch.')
53-
parser.add_argument('--data_dir', type=str, default='/tmp/cifar10_data', help='Path to the CIFAR-10 data directory.')
54-
parser.add_argument('--use_fp16', type=bool, default=False, help='Train the model using fp16.')
52+
parser.add_argument('--batch_size', type=int, default=128,
53+
help='Number of images to process in a batch.')
54+
55+
parser.add_argument('--data_dir', type=str, default='/tmp/cifar10_data',
56+
help='Path to the CIFAR-10 data directory.')
57+
58+
parser.add_argument('--use_fp16', type=bool, default=False,
59+
help='Train the model using fp16.')
5560

5661
FLAGS = parser.parse_args()
5762
# Global constants describing the CIFAR-10 data set.

tutorials/image/cifar10/cifar10_eval.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -46,17 +46,23 @@
4646

4747
parser = cifar10.parser
4848

49-
parser.add_argument('--eval_dir', type=str, default='/tmp/cifar10_eval', help='Directory where to write event logs.')
49+
parser.add_argument('--eval_dir', type=str, default='/tmp/cifar10_eval',
50+
help='Directory where to write event logs.')
5051

51-
parser.add_argument('--eval_data', type=str, default='test', help='Either `test` or `train_eval`.')
52+
parser.add_argument('--eval_data', type=str, default='test',
53+
help='Either `test` or `train_eval`.')
5254

53-
parser.add_argument('--checkpoint_dir', type=str, default='/tmp/cifar10_train', help='Directory where to read model checkpoints.')
55+
parser.add_argument('--checkpoint_dir', type=str, default='/tmp/cifar10_train',
56+
help='Directory where to read model checkpoints.')
5457

55-
parser.add_argument('--eval_interval_secs', type=int, default=60*5, help='How often to run the eval.')
58+
parser.add_argument('--eval_interval_secs', type=int, default=60*5,
59+
help='How often to run the eval.')
5660

57-
parser.add_argument('--num_examples', type=int, default=10000, help='Number of examples to run.')
61+
parser.add_argument('--num_examples', type=int, default=10000,
62+
help='Number of examples to run.')
5863

59-
parser.add_argument('--run_once', type=bool, default=False, help='Whether to run eval only once.')
64+
parser.add_argument('--run_once', type=bool, default=False,
65+
help='Whether to run eval only once.')
6066

6167

6268
def eval_once(saver, summary_writer, top_k_op, summary_op):

tutorials/image/cifar10/cifar10_multi_gpu_train.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -52,13 +52,17 @@
5252

5353
parser = cifar10.parser
5454

55-
parser.add_argument('--train_dir', type=str, default='/tmp/cifar10_train', help='Directory where to write event logs and checkpoint.')
55+
parser.add_argument('--train_dir', type=str, default='/tmp/cifar10_train',
56+
help='Directory where to write event logs and checkpoint.')
5657

57-
parser.add_argument('--max_steps', type=int, default=1000000, help='Number of batches to run.')
58+
parser.add_argument('--max_steps', type=int, default=1000000,
59+
help='Number of batches to run.')
5860

59-
parser.add_argument('--num_gpus', type=int, default=1, help='How many GPUs to use.')
61+
parser.add_argument('--num_gpus', type=int, default=1,
62+
help='How many GPUs to use.')
6063

61-
parser.add_argument('--log_device_placement', type=bool, default=False, help='Whether to log device placement.')
64+
parser.add_argument('--log_device_placement', type=bool, default=False,
65+
help='Whether to log device placement.')
6266

6367

6468
def tower_loss(scope, images, labels):

tutorials/image/cifar10/cifar10_train.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,13 +46,17 @@
4646

4747
parser = cifar10.parser
4848

49-
parser.add_argument('--train_dir', type=str, default='/tmp/cifar10_train', help='Directory where to write event logs and checkpoint.')
49+
parser.add_argument('--train_dir', type=str, default='/tmp/cifar10_train',
50+
help='Directory where to write event logs and checkpoint.')
5051

51-
parser.add_argument('--max_steps', type=int, default=1000000, help='Number of batches to run.')
52+
parser.add_argument('--max_steps', type=int, default=1000000,
53+
help='Number of batches to run.')
5254

53-
parser.add_argument('--log_device_placement', type=bool, default=False, help='Whether to log device placement.')
55+
parser.add_argument('--log_device_placement', type=bool, default=False,
56+
help='Whether to log device placement.')
5457

55-
parser.add_argument('--log_frequency', type=int, default=10, help='How often to log results to the console.')
58+
parser.add_argument('--log_frequency', type=int, default=10,
59+
help='How often to log results to the console.')
5660

5761

5862
def train():

0 commit comments

Comments
 (0)