diff options
Diffstat (limited to 'estimator_ckpt_converter')
-rwxr-xr-x | estimator_ckpt_converter | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/estimator_ckpt_converter b/estimator_ckpt_converter new file mode 100755 index 0000000..6d357cb --- /dev/null +++ b/estimator_ckpt_converter @@ -0,0 +1,8 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +import re +import sys +from tensorflow_estimator.python.estimator.tools.checkpoint_converter import main +if __name__ == '__main__': + sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) + sys.exit(main()) |