-
Notifications
You must be signed in to change notification settings - Fork 2
/
config.py
22 lines (19 loc) · 913 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import tensorflow as tf
FLAGS = tf.app.flags.FLAGS
tf.app.flags.DEFINE_integer('raw_height', 144, '')
tf.app.flags.DEFINE_integer('raw_width', 144, '')
tf.app.flags.DEFINE_integer('height', 128, '')
tf.app.flags.DEFINE_integer('width', 128, '')
tf.app.flags.DEFINE_integer('depth', 3, '')
tf.app.flags.DEFINE_integer('num_class', 3, '')
tf.app.flags.DEFINE_integer('num_conv', 2, '')
tf.app.flags.DEFINE_integer('kernel_size', 5, '')
tf.app.flags.DEFINE_integer('pool_size', 2, '')
tf.app.flags.DEFINE_integer('num_map', 32, '')
tf.app.flags.DEFINE_integer('num_fc_layer', 2, '')
tf.app.flags.DEFINE_integer('num_fc_input', 512, '')
tf.app.flags.DEFINE_integer('max_steps', 1000000, '')
tf.app.flags.DEFINE_integer('batch_size', 32, '')
tf.app.flags.DEFINE_float('learning_rate', 0.0001, '')
tf.app.flags.DEFINE_string('data_dir', './data', '')
tf.app.flags.DEFINE_string('checkpoint_dir', './checkpoint', '')