From 878e79a6d8218e030a516cfc8d5487bcd1cd8804 Mon Sep 17 00:00:00 2001 From: Jen Iofinova Date: Wed, 2 Aug 2023 20:46:08 +0000 Subject: [PATCH] make quality --- src/sparseml/pytorch/torchvision/train.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/sparseml/pytorch/torchvision/train.py b/src/sparseml/pytorch/torchvision/train.py index 747348c7f1d..101c08ae842 100644 --- a/src/sparseml/pytorch/torchvision/train.py +++ b/src/sparseml/pytorch/torchvision/train.py @@ -772,7 +772,6 @@ def _create_model( num_classes=num_classes, ) - if isinstance(model, tuple): model, arch_key = model elif arch_key in torchvision.models.__dict__: