Skip to content

Y
ygo-agent
Commit 03416f14 authored by sbl1996@126.com's avatar sbl1996@126.com
Browse files
Options

Fix

parent a9b19128
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 7 deletions
scripts/cleanba.py
View file @ 03416f14
...@@ -1190,13 +1190,6 @@ def main(): ...@@ -1190,13 +1190,6 @@ def main():
M_steps = tb_global_step // 2**20 M_steps = tb_global_step // 2**20
ckpt_name = f"{timestamp}_{M_steps}M.flax_model" ckpt_name = f"{timestamp}_{M_steps}M.flax_model"
ckpt_maneger.save(unreplicated_params, ckpt_name) ckpt_maneger.save(unreplicated_params, ckpt_name)
if args.gcs_bucket is not None:
lastest_path = ckpt_maneger.get_latest()
copy_path = lastest_path.with_name("latest" + lastest_path.suffix)
shutil.copyfile(lastest_path, copy_path)
zip_file_path = "latest.zip"
zip_files(zip_file_path, [str(copy_path), tb_log_dir])
sync_to_gcs(args.gcs_bucket, zip_file_path)
if learner_policy_version >= args.num_updates: if learner_policy_version >= args.num_updates:
break break
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment