diff --git a/.vscode/launch.json b/.vscode/launch.json index cf9abdc..bc3cb5f 100644 --- a/.vscode/launch.json +++ b/.vscode/launch.json @@ -17,7 +17,7 @@ // "--ckpt_path", // "${workspaceFolder}/lightning_logs/version_264/checkpoints/epoch=9-st&ep=1000.ckpt", "--data.num_workers", - "0", + "1", "--trainer.benchmark", "false", "--trainer.num_sanity_val_steps", diff --git a/README.md b/README.md index 5408c5b..e0ff277 100644 --- a/README.md +++ b/README.md @@ -28,6 +28,17 @@ Start inference on images: python src/main.py predict --ckpt_path ``` +Quick and dirty way to export to `.onnx`: +```python +>>> from src.module import DETR +>>> checkpoint = "" +>>> model = DETR.load_from_checkpoint(checkpoint) +>>> model.net.save_pretrained("hugginface_checkpoint") +``` +```bash +python -m transformers.onnx --model=hugginface_checkpoint onnx_export/ +``` + ## License Distributed under the [MIT](https://choosealicense.com/licenses/mit/) license. \