enable external logging boolean correctly

model.py

@@ -151,7 +151,7 @@ class SmolLM3Model:
             # Only enable DDP if multiple GPUs are available
             "ddp_backend": self.config.ddp_backend if torch.cuda.device_count() > 1 else None,
             "ddp_find_unused_parameters": self.config.ddp_find_unused_parameters if torch.cuda.device_count() > 1 else False,
-            "report_to": "none",  # Disable external logging
+            "report_to": None,  # Enable external logging (default)
             "remove_unused_columns": False,
             "dataloader_pin_memory": False,
             "group_by_length": True,
@@ -172,17 +172,7 @@ class SmolLM3Model:
         # Override with kwargs
         training_args.update(kwargs)
         
-        # Debug: Check for any boolean values that might be causing issues
-        for key, value in training_args.items():
-            if isinstance(value, bool):
-                logger.info(f"Boolean argument: {key} = {value}")
-        
-        try:
-            return TrainingArguments(**training_args)
-        except Exception as e:
-            logger.error(f"Failed to create TrainingArguments: {e}")
-            logger.error(f"Training arguments: {training_args}")
-            raise
+        return TrainingArguments(**training_args)
     
     def save_pretrained(self, path: str):
         """Save model and tokenizer"""

trainer.py

@@ -98,19 +98,18 @@ class SmolLM3Trainer:
         callbacks.append(SimpleConsoleCallback())
         logger.info("Added simple console monitoring callback")
         
-        # Try to add Trackio callback if available (temporarily disabled for debugging)
-        logger.info("Skipping Trackio callback to debug training issue")
-        # if self.monitor and self.monitor.enable_tracking:
-        #     try:
-        #         trackio_callback = self.monitor.create_monitoring_callback()
-        #         if trackio_callback:
-        #         callbacks.append(trackio_callback)
-        #         logger.info("Added Trackio monitoring callback")
-        #     else:
-        #         logger.warning("Failed to create Trackio callback")
-        # except Exception as e:
-        #     logger.error(f"Error creating Trackio callback: {e}")
-        #     logger.info("Continuing with console monitoring only")
+        # Try to add Trackio callback if available
+        if self.monitor and self.monitor.enable_tracking:
+            try:
+                trackio_callback = self.monitor.create_monitoring_callback()
+                if trackio_callback:
+                    callbacks.append(trackio_callback)
+                    logger.info("Added Trackio monitoring callback")
+                else:
+                    logger.warning("Failed to create Trackio callback")
+            except Exception as e:
+                logger.error(f"Error creating Trackio callback: {e}")
+                logger.info("Continuing with console monitoring only")
         
         # Try standard Trainer first (more stable with callbacks)
         try: