fix: losses take item

Author: AlphaPav

runners/atlasnet_runner.py

@@ -50,7 +50,7 @@ def train_step(self, items):
 
         self.loss["refine_loss"] = refine_loss * 1000
         self.loss["rec_loss"] = _loss
-        self.losses.update([refine_loss * 1000])
+        self.losses.update([refine_loss.item() * 1000])
 
     def val_step(self, items):
         _, (_, _, _, data) = items

runners/grnet_runner.py

@@ -50,7 +50,7 @@ def train_step(self, items):
         self.loss["coarse_loss"] = coarse_loss * 1000
         self.loss["refine_loss"] = refine_loss * 1000
         self.loss["rec_loss"] = _loss
-        self.losses.update([coarse_loss * 1000, refine_loss * 1000])
+        self.losses.update([coarse_loss.item() * 1000, refine_loss.item() * 1000])
 
     def val_step(self, items):
         _, (_, _, _, data) = items

runners/msn_runner.py

@@ -46,7 +46,7 @@ def train_step(self, items):
         self.loss["coarse_loss"] = coarse_loss * 1000
         self.loss["refine_loss"] = refine_loss * 1000
         self.loss["rec_loss"] = _loss
-        self.losses.update([coarse_loss * 1000, refine_loss * 1000])
+        self.losses.update([coarse_loss.item() * 1000, refine_loss.item() * 1000])
 
     def val_step(self, items):
         _, (_, _, _, data) = items

runners/sparenet_runner.py

@@ -52,7 +52,7 @@ def train_step(self, items):
         self.loss["coarse_loss"] = coarse_loss * 1000
         self.loss["refine_loss"] = refine_loss * 1000
         self.loss["rec_loss"] = _loss
-        self.losses.update([coarse_loss * 1000, refine_loss * 1000])
+        self.losses.update([coarse_loss.item() * 1000, refine_loss.item() * 1000])
 
     def val_step(self, items):
         _, (_, _, _, data) = items