sparseml
sparseml copied to clipboard
Create deployment folder for IC integration
Testing
python export.py
--dataset
imagenette
--dataset-path
/home/damian/imagenette
--checkpoint-path
pytorch_vision/mobilenet_v1_imagenette__03/training/checkpoint-best.pth
results in
Model id is set to _imagenette__16
already downloaded imagenette ImagenetteSize.s160
2022-08-11 06:54:56 __main__ INFO exporting pytorch in /home/damian/sparseml/pytorch_vision/_imagenette__16
2022-08-11 06:54:56 __main__ INFO exporting pytorch in /home/damian/sparseml/pytorch_vision/_imagenette__16
Exporting samples: 0%| | 0/1 [00:00<?, ?it/s]2022-08-11 06:54:56 __main__ INFO exporting onnx in /home/damian/sparseml/pytorch_vision/_imagenette__16
2022-08-11 06:54:56 __main__ INFO exporting onnx in /home/damian/sparseml/pytorch_vision/_imagenette__16
Exporting samples: 100%|██████████████████████████| 1/1 [00:00<00:00, 1.32it/s]
2022-08-11 06:54:57 sparseml.pytorch.utils.exporter INFO Created deployment folder at /home/damian/sparseml/pytorch_vision/_imagenette__16/deployment
2022-08-11 06:54:57 sparseml.pytorch.utils.exporter INFO Created saving model.onnx at deployment folder at /home/damian/sparseml/pytorch_vision/_imagenette__16/deployment/model.onnx
and
damian@lambdaquad:~/sparseml/pytorch_vision/_imagenette__16/deployment$ ls
>> model.onnx
@bfineran assigned for review