Update lenet.py

haebeom-lee · Jan 26, 2019 · e9379fa · e9379fa
1 parent 4c68594
commit e9379fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lenet.py b/lenet.py
@@ -55,5 +55,5 @@ def dropmax(x, y, training, name='dropmax', reuse=None):
     # dropmax modules
     net['kl'] = kl_divergence(p, q, y)
     net['aux'] = auxloss(r, y)
-    net['ent'] = entropy(p)
+    net['neg_ent'] = neg_entropy(p)
     return net