Merge pull request #2685 from juandspy/patch-1

Update TRAINING.rst  (mmap-able model)
This commit is contained in:
lissyx 2020-01-22 16:58:55 +01:00 committed by GitHub
commit a9855f1e4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -146,6 +146,11 @@ def main():
maybe_download_tc(target_dir=args.target, tc_url=get_tc_url(args.arch, args.artifact, args.branch)) maybe_download_tc(target_dir=args.target, tc_url=get_tc_url(args.arch, args.artifact, args.branch))
if args.artifact == "convert_graphdef_memmapped_format":
convert_graph_file = os.path.join(args.target, args.artifact)
final_stat = os.stat(convert_graph_file)
os.chmod(convert_graph_file, final_stat.st_mode | stat.S_IEXEC)
if '.tar.' in args.artifact: if '.tar.' in args.artifact:
subprocess.check_call(['tar', 'xvf', os.path.join(args.target, args.artifact), '-C', args.target]) subprocess.check_call(['tar', 'xvf', os.path.join(args.target, args.artifact), '-C', args.target])