Skip to content
Snippets Groups Projects
Commit c0ca45c7 authored by Nikos Athanasiou's avatar Nikos Athanasiou
Browse files

exps cluster

parent 479eba74
Branches
No related tags found
No related merge requests found
......@@ -18,22 +18,17 @@ def run(cmd):
# teach/independent sampling
base_dir = Path("/is/cluster/work/nathanasiou/experiments/teach/babel-amass/teach-post-submission")
# temos/teach final exps
models = ['teach', 'temos']
motion_branch = ['true', 'false']
motion_branch = ['true']
teacher_forcing = ['true', 'false']
hist_frames= [1,2,5,10]
for hf in hist_frames:
for is_motion_branch in motion_branch:
for tf in teacher_forcing:
if hf == 5 or hf == 10 and is_motion_branch == 'false':
continue
else:
cmd = ['python', 'cluster/single_run.py', '--expname', 'teach-post-submission', '--mode', 'train', '--bid', '20']
cmd.extend(['--run-id', f'teach-hf-{hf}-motion_branch_{is_motion_branch}-teacher_force_{tf}', '--extras',
f"model.teacher_forcing={tf} model.hist_frames={hf} model.motion_branch={is_motion_branch} data.dtype=separate_pairs callback.render.every_n_epochs=50 model=teach"])
cmd.extend(['--run-id', f'teach-hf-{hf}-motion_branch_true-teacher_force_{tf}', '--extras',
f"model.teacher_forcing={tf} model.hist_frames={hf} model.motion_branch=true data.dtype=separate_pairs callback.render.every_n_epochs=50 model=teach"])
run(cmd)
end_script()
# exps_to_sample = ['teach-motion_branch_false-teacher_force_false',
# 'teach-motion_branch_false-teacher_force_true',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment