Skip to content

Commit

Permalink
Make enet to work by fixing typo
Browse files Browse the repository at this point in the history
  • Loading branch information
whchung committed Apr 2, 2018
1 parent bc3e88d commit 42e5562
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 42e5562

Please sign in to comment.