-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathrun_shell_blender.py
25 lines (24 loc) · 1.16 KB
/
run_shell_blender.py
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
import os
iterations = 30_000 # or 40_000, or at your will
for cuda, scene in enumerate(['chair', 'drums', 'ficus', 'hotdog', 'lego', 'materials', 'mic', 'ship']):
one_cmd = (f'CUDA_VISIBLE_DEVICES={0} python train.py '
f'-s data/nerf_synthetic/{scene} '
f'--eval '
f'--lod 0 '
f'--voxel_size 0.001 '
f'--update_init_factor 4 '
f'-m output/nerf_synthetic/{scene} '
f'--lmbda_list 4 2 0.25 '
f'--iterations {iterations} '
f'--position_lr_max_steps {iterations} '
f'--offset_lr_max_steps {iterations} '
f'--mask_lr_max_steps {iterations} '
f'--mlp_opacity_lr_max_steps {iterations} '
f'--mlp_cov_lr_max_steps {iterations} '
f'--mlp_color_lr_max_steps {iterations} '
f'--mlp_featurebank_lr_max_steps {iterations} '
f'--encoding_xyz_lr_max_steps {iterations} '
f'--mlp_grid_lr_max_steps {iterations} '
f'--mlp_deform_lr_max_steps {iterations} '
f'--mask_lr_final {0.00005}')
os.system(one_cmd)