summaryrefslogtreecommitdiffstats
path: root/estimator_ckpt_converter
diff options
context:
space:
mode:
Diffstat (limited to 'estimator_ckpt_converter')
-rwxr-xr-xestimator_ckpt_converter8
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())