Update demo.py

drahnreb · Aug 20, 2019 · 6749f2e · 6749f2e
1 parent 21a57cc
commit 6749f2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.py b/demo.py
@@ -17,7 +17,7 @@
 from torchvision import datasets, transforms
 
 # import Mila activation function
-from Mish.Torch.mish import mish
+from Mish.Torch.mish import Mish as mish
 import Mish.Torch.functional as Func
 
 # activation names constant