| #!/usr/bin/env python3 | |
| import sys,re | |
| # args: base_config out_config stage1_cache out_root ds_name exp views schedules_csv | |
| base,out,stage1_cache,out_root,ds_name,exp,views,sched = sys.argv[1:9] | |
| txt=open(base).read() | |
| # cut everything from 'val_appforce:' to EOF, replace with our block | |
| idx=txt.index('\nval_appforce:') | |
| head=txt[:idx] | |
| sched_list='['+', '.join(sched.split(','))+']' | |
| block=f""" | |
| val_appforce: | |
| enabled: true | |
| every: 5000 | |
| at_step0: false | |
| views: {views} | |
| schedules: {sched_list} | |
| seed: 42 | |
| limit: 0 | |
| batch_size: 6 | |
| single: true | |
| decode_procs: 1 | |
| ref_threads: 8 | |
| stage1_cache: {stage1_cache} | |
| out_root: {out_root} | |
| ref_python: /lp-dev/jonghoon/mv-mesh/envs/mv-sam3d/bin/python | |
| eval_dataset_flag: toys4k | |
| datasets: | |
| - name: {ds_name} | |
| exp: {exp} | |
| """ | |
| open(out,'w').write(head+block) | |
| print("wrote",out) | |