sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard

Create deployment folder for IC integration

Open dbogunowicz opened this issue 3 years ago • 1 comments

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

dbogunowicz avatar Aug 11 '22 10:08 dbogunowicz

@bfineran assigned for review

github-actions[bot] avatar Aug 11 '22 10:08 github-actions[bot]