allow int or float loss weight type

Author: wangguan1995

ppsci/loss/integral.py

@@ -71,7 +71,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
             elif self.reduction == "mean":
                 loss = loss.mean()
 
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]

ppsci/loss/l1.py

@@ -59,7 +59,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
             loss = F.l1_loss(output_dict[key], label_dict[key], "none")
             if weight_dict:
                 loss *= weight_dict[key]
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]
@@ -72,7 +72,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
             elif self.reduction == "mean":
                 loss = loss.mean()
 
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]
@@ -122,7 +122,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
             elif self.reduction == "mean":
                 loss = loss.mean()
 
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]

ppsci/loss/l2.py

@@ -52,7 +52,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
                 loss *= output_dict["area"]
 
             loss = loss.sum()
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]
@@ -94,7 +94,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
                 loss *= output_dict["area"]
 
             loss = loss.sum()
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]

ppsci/loss/mse.py

@@ -68,7 +68,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
                 loss = loss.sum()
             elif self.reduction == "mean":
                 loss = loss.mean()
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]
@@ -168,7 +168,7 @@ def forward(self, output_dict, label_dict, weight_dict=None):
             elif self.reduction == "mean":
                 loss = loss.mean()
 
-            if isinstance(self.weight, float):
+            if isinstance(self.weight, (float, int)):
                 loss *= self.weight
             elif isinstance(self.weight, dict) and key in self.weight:
                 loss *= self.weight[key]