Skip to content
Browse files

ONNX export. Fix #61

  • Loading branch information...
ocourtin committed Aug 8, 2019
1 parent baf4a0f commit 63cc2a9774b03fcba4d8cfa37e184fa95daee7cc
Showing with 1 addition and 1 deletion.
  1. +1 −1 Makefile
@@ -85,7 +85,7 @@ it_train:
@echo "==================================================================================="
@rsp export --checkpoint it/pth/checkpoint-00003.pth --type jit it/pth/export.jit
#@rsp export --checkpoint it/pth/checkpoint-00003.pth --type onnx it/pth/export.onnx
@rsp export --checkpoint it/pth/checkpoint-00003.pth --type onnx it/pth/export.onnx
@rsp predict --config config.toml --bs 4 --checkpoint it/pth/checkpoint-00003.pth it/prediction it/prediction/masks
@rsp compare --images it/prediction/images it/prediction/labels it/prediction/masks --mode stack --labels it/prediction/labels --masks it/prediction/masks it/prediction/compare
@rsp compare --images it/prediction/images it/prediction/compare --mode side it/prediction/compare_side

0 comments on commit 63cc2a9

Please sign in to comment.
You can’t perform that action at this time.