Merge pull request #2685 from juandspy/patch-1
Update TRAINING.rst (mmap-able model)
This commit is contained in:
commit
a9855f1e4e
@ -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])
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user