Fix referenced before assignment error (#1340)

abhinavnmagic · bfineran · commit f7cc6a6425d3 · 2023-02-03T10:26:28.000-05:00
diff --git a/src/sparseml/pytorch/torchvision/train.py b/src/sparseml/pytorch/torchvision/train.py
@@ -389,6 +389,8 @@ def collate_fn(batch):
             device=device,
             num_classes=num_classes,
         )
+    else:
+        distill_teacher = args.distill_teacher
 
     if args.distributed and args.sync_bn:
         model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model)

Original file line number	Diff line number	Diff line change
`@@ -389,6 +389,8 @@ def collate_fn(batch):`
`389`	`389`	`device=device,`
`390`	`390`	`num_classes=num_classes,`
`391`	`391`	`)`
	`392`	`+ else:`
	`393`	`+ distill_teacher = args.distill_teacher`
`392`	`394`
`393`	`395`	`if args.distributed and args.sync_bn:`
`394`	`396`	`model = torch.nn.SyncBatchNorm.convert_sync_batchnorm(model)`