Skip to content

S
seminar-breakout
Commit bac11ae3 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix examples

parent 67fdbf26
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 12 deletions
example_alexnet.py
View file @ bac11ae3
......@@ -24,12 +24,8 @@ CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE
class Model(ModelDesc):
def _get_input_vars(self):
return [
tf.placeholder(
tf.float32, shape=(None, 227, 227, 3), name='input'),
tf.placeholder(
tf.int32, shape=(None,), name='label')
]
return [InputVar(tf.float32, (None, 227, 227, 3), 'input'),
InputVar(tf.int32, (None,), 'label') ]
def _get_cost(self, inputs, is_training):
# img: 227x227x3
......
example_cifar10.py
View file @ bac11ae3
......@@ -27,11 +27,8 @@ CAPACITY = MIN_AFTER_DEQUEUE + 3 * BATCH_SIZE
class Model(ModelDesc):
def _get_input_vars(self):
return [
tf.placeholder(
tf.float32, shape=[None, 30, 30, 3], name='input'),
tf.placeholder(
tf.int32, shape=[None], name='label')
return [InputVar(tf.float32, [None, 30, 30, 3], 'input'),
InputVar(tf.int32, [None], 'label')
]
def _get_cost(self, input_vars, is_training):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment