Skip to content

Commit

Permalink
set train = true
Browse files Browse the repository at this point in the history
  • Loading branch information
amirbar committed Jun 5, 2017
1 parent 2e5597a commit 9c1e56f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train/train.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -127,7 +127,7 @@ def train():

## network
logits, end_points, pyramid_map = network.get_network(FLAGS.network, image,
weight_decay=FLAGS.weight_decay)
weight_decay=FLAGS.weight_decay, is_training=True)
outputs = pyramid_network.build(end_points, im_shape[1], im_shape[2], pyramid_map,
num_classes=81,
base_anchors=9,
Expand Down

1 comment on commit 9c1e56f

@souryuu
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thank you for the update. This part is crucial to get a decent classification results from refined network.

Please sign in to comment.