Fix SmolVLA loss not sent to wandb (#1198)

lerobot/common/policies/smolvla/modeling_smolvla.py

@@ -338,7 +338,7 @@ class SmolVLAPolicy(PreTrainedPolicy):
         # For backward pass
         loss = losses.mean()
         # For backward pass
-        loss_dict["loss"] = loss
+        loss_dict["loss"] = loss.item()
         return loss, loss_dict
 
     def prepare_images(self, batch):

lerobot/common/utils/wandb_utils.py

@@ -115,7 +115,7 @@ class WandBLogger:
         for k, v in d.items():
             if not isinstance(v, (int, float, str)):
                 logging.warning(
-                    f'WandB logging of key "{k}" was ignored as its type is not handled by this wrapper.'
+                    f'WandB logging of key "{k}" was ignored as its type "{type(v)}" is not handled by this wrapper.'
                 )
                 continue
             self._wandb.log({f"{mode}/{k}": v}, step=step)