removing checkpointing recipe if its None (#1308)

abhinavnmagic · bfineran · commit a83a67ad712d · 2023-02-03T10:26:28.000-05:00
diff --git a/src/sparseml/pytorch/torchvision/train.py b/src/sparseml/pytorch/torchvision/train.py
@@ -467,7 +467,7 @@ def collate_fn(batch):
         )
         checkpoint_manager = (
             ScheduledModifierManager.from_yaml(checkpoint["recipe"])
-            if "recipe" in checkpoint
+            if "recipe" in checkpoint and checkpoint["recipe"] is not None
             else None
         )
     elif args.resume:
@@ -635,7 +635,8 @@ def log_metrics(tag: str, metrics: utils.MetricLogger, epoch: int, epoch_step: i
                 )
             else:
                 checkpoint["epoch"] = -1 if epoch == max_epochs - 1 else epoch
-                checkpoint["recipe"] = str(manager)
+                if str(manager) is not None:
+                    checkpoint["recipe"] = str(manager)
 
             file_names = ["checkpoint.pth"]
             if is_new_best:

Original file line number	Diff line number	Diff line change
`@@ -467,7 +467,7 @@ def collate_fn(batch):`
`467`	`467`	`)`
`468`	`468`	`checkpoint_manager = (`
`469`	`469`	`ScheduledModifierManager.from_yaml(checkpoint["recipe"])`
`470`		`- if "recipe" in checkpoint`
	`470`	`+ if "recipe" in checkpoint and checkpoint["recipe"] is not None`
`471`	`471`	`else None`
`472`	`472`	`)`
`473`	`473`	`elif args.resume:`
`@@ -635,7 +635,8 @@ def log_metrics(tag: str, metrics: utils.MetricLogger, epoch: int, epoch_step: i`
`635`	`635`	`)`
`636`	`636`	`else:`
`637`	`637`	`checkpoint["epoch"] = -1 if epoch == max_epochs - 1 else epoch`
`638`		`- checkpoint["recipe"] = str(manager)`
	`638`	`+ if str(manager) is not None:`
	`639`	`+ checkpoint["recipe"] = str(manager)`
`639`	`640`
`640`	`641`	`file_names = ["checkpoint.pth"]`
`641`	`642`	`if is_new_best:`