added class method

madhavkhoslaa · Aug 24, 2019 · 9746b28 · 9746b28
1 parent 9b9bc49
commit 9746b28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loss/diceloss.py b/loss/diceloss.py
@@ -5,7 +5,7 @@
 
 class Loss:
     @classmethod
-    def dice_loss(input, target):
+    def dice_loss(cls, input, target):
         smooth = 1.
         iflat = input.view(-1)
         tflat = target.view(-1)