Show DeMemWM training loss in progress bar

algorithms/dememwm/df_video.py

@@ -596,7 +596,7 @@ class DeMemWMMinecraft(DiffusionForcingBase):
                 loss = self.reweight_loss(loss, None)
 
             if batch_idx % 20 == 0:
-                self.log("training/loss", loss.cpu())
+                self.log("training/loss", loss.detach(), prog_bar=True, sync_dist=True)
 
             return {"loss": loss}
         raise TypeError(