Merge pull request #16 from MihailSalnikov/patch-1

supporting python3 for pytorch
bermanmaxim · Nov 19, 2018 · 4b0e391 · 4b0e391
2 parents c74451a + 8e49005
commit 4b0e391
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch/lovasz_losses.py b/pytorch/lovasz_losses.py
@@ -12,7 +12,7 @@
 try:
     from itertools import  ifilterfalse
 except ImportError: # py3k
-    from itertools import  filterfalse
+    from itertools import  filterfalse as ifilterfalse
 
 
 def lovasz_grad(gt_sorted):