Skip to content
This repository was archived by the owner on Jun 3, 2025. It is now read-only.

Commit ea432e9

Browse files
authored
Remove pytorch export from IC export script (#1005)
1 parent 1e5b27b commit ea432e9

File tree

1 file changed

+2
-8
lines changed
  • src/sparseml/pytorch/image_classification

1 file changed

+2
-8
lines changed

src/sparseml/pytorch/image_classification/export.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -387,18 +387,12 @@ def export(
387387
"""
388388
exporter = ModuleExporter(model, save_dir)
389389

390-
# export PyTorch state dict
391-
LOGGER.info(f"exporting pytorch in {save_dir}")
392-
393-
exporter.export_pytorch(
394-
use_zipfile_serialization_if_available=(use_zipfile_serialization_if_available)
395-
)
396-
onnx_exported = False
397-
398390
if not val_loader:
399391
# create fake data for export
400392
val_loader = [[torch.randn(1, 3, image_size, image_size)]]
401393

394+
onnx_exported = False
395+
402396
for batch, data in tqdm(
403397
enumerate(val_loader),
404398
desc="Exporting samples",

0 commit comments

Comments
 (0)