Skip to content

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

add a missing output in dorefa (fix #800)

parent 69465fc7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/DoReFa-Net/alexnet-dorefa.py
View file @ c0ae913e
...@@ -128,6 +128,7 @@ class Model(ImageNetModel): ...@@ -128,6 +128,7 @@ class Model(ImageNetModel):
.apply(nonlin) .apply(nonlin)
.FullyConnected('fct', 1000, use_bias=True)()) .FullyConnected('fct', 1000, use_bias=True)())
add_param_summary(('.*/W', ['histogram', 'rms'])) add_param_summary(('.*/W', ['histogram', 'rms']))
tf.nn.softmax(logits, name='output') # for prediction
return logits return logits
def optimizer(self): def optimizer(self):
......
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