Adding config to wandb logger and calling LoggerManager.save(recipe) (#1318)

Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>

Author: 2 people

src/sparseml/pytorch/torchvision/train.py

@@ -545,13 +545,19 @@ def collate_fn(batch):
             TensorBoardLogger(log_path=args.output_dir),
         ]
         try:
-            loggers.append(WANDBLogger())
+            config = vars(args)
+            if manager is not None:
+                config["manager"] = str(manager)
+            loggers.append(WANDBLogger(init_kwargs=dict(config=config)))
         except ImportError:
             warnings.warn("Unable to import wandb for logging")
         logger = LoggerManager(loggers)
     else:
         logger = LoggerManager(log_python=False)
 
+    if args.recipe is not None:
+        logger.save(args.recipe)
+
     steps_per_epoch = len(data_loader) / args.gradient_accum_steps
 
     def log_metrics(tag: str, metrics: utils.MetricLogger, epoch: int, epoch_step: int):