Remove pytorch export from IC export script (#1005)

neuralmagic · Aug 18, 2022 · ea432e9 · ea432e9
1 parent 1e5b27b
commit ea432e9
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/src/sparseml/pytorch/image_classification/export.py b/src/sparseml/pytorch/image_classification/export.py
@@ -387,18 +387,12 @@ def export(
     """
     exporter = ModuleExporter(model, save_dir)
 
-    # export PyTorch state dict
-    LOGGER.info(f"exporting pytorch in {save_dir}")
-
-    exporter.export_pytorch(
-        use_zipfile_serialization_if_available=(use_zipfile_serialization_if_available)
-    )
-    onnx_exported = False
-
     if not val_loader:
         # create fake data for export
         val_loader = [[torch.randn(1, 3, image_size, image_size)]]
 
+    onnx_exported = False
+
     for batch, data in tqdm(
         enumerate(val_loader),
         desc="Exporting samples",