File size: 1,358 Bytes
c985ba4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
import os
import shutil
def cp_ckpt(remote_dir="data_wd/youtube_vos_jobs/result", curr_dir="backup"):
exps = os.listdir(curr_dir)
for exp in exps:
print("Exp: ", exp)
exp_dir = os.path.join(curr_dir, exp)
stages = os.listdir(exp_dir)
for stage in stages:
print("Stage: ", stage)
stage_dir = os.path.join(exp_dir, stage)
finals = ["ema_ckpt", "ckpt"]
for final in finals:
print("Final: ", final)
final_dir = os.path.join(stage_dir, final)
ckpts = os.listdir(final_dir)
for ckpt in ckpts:
if '.pth' not in ckpt:
continue
curr_ckpt_path = os.path.join(final_dir, ckpt)
remote_ckpt_path = os.path.join(remote_dir, exp, stage,
final, ckpt)
if os.path.exists(remote_ckpt_path):
os.system('rm {}'.format(remote_ckpt_path))
try:
shutil.copy(curr_ckpt_path, remote_ckpt_path)
print(ckpt, ': OK')
except OSError as Inst:
print(Inst)
print(ckpt, ': Fail')
if __name__ == "__main__":
cp_ckpt()
|