Skip to content

Commit

Permalink
Merge pull request #7 from whchung/master
Browse files Browse the repository at this point in the history
Make enet to work by fixing typo
  • Loading branch information
hellochick authored Apr 3, 2018
2 parents bc3e88d + 42e5562 commit 2040423
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion model.py
Original file line number Diff line number Diff line change
Expand Up @@ -895,7 +895,7 @@ def get_variable_weight_decay(self, name, shape, initializer, loss_category,
weight_decay = self.wd*tf.nn.l2_loss(variable)
tf.add_to_collection(loss_category, weight_decay)

return variabl
return variable

class ICNet(Network):
def __init__(self, is_training=False, num_classes=19, input_size=[1024, 2048]):
Expand Down

0 comments on commit 2040423

Please sign in to comment.