-
Notifications
You must be signed in to change notification settings - Fork 1
/
run_s2_wmask.sh
54 lines (47 loc) · 1.58 KB
/
run_s2_wmask.sh
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
code_path=/data/code/nefii_v0.1/code
data_path=/data/datasets/nefii/ds_physg/
save_path=/data/datasets/nefii/Experiments/20230710_wmask_sphere/
cd $code_path
source /root/anaconda3/etc/profile.d/conda.sh
conda activate nefii
Scene=thin_cube
geometry_neus=/data/datasets/nefii/Experiments/20230710_wmask_sphere/thin_cube/checkpoints/ckpt_300000.pth
#DDP
CUDA_VISIBLE_DEVICES=0,1,2,3 python -u -m torch.distributed.launch --nproc_per_node 4 \
$code_path/training/exp_runner.py \
--conf $code_path/confs_sg/conf_neus.conf \
--data_split_dir $data_path/$Scene/train/ \
--data_split_dir_test $data_path/$Scene/test/ \
--exps_folder_name $save_path \
--expname 05_unknow_$Scene \
--nepoch 2000 \
--max_niter 200001 \
--gamma 2.2 \
--batch_size 1 \
--roughness_warmup 5000 \
--coordinate_type blender \
--memory_capacity_level 15 \
--secondary_batch_size 1024 \
--secondary_train_interval 10 \
--freeze_geometry \
--geometry_neus $geometry_neus\
2>&1 | tee run_00.log
# python \
# $code_path/training/exp_runner.py \
# --conf $code_path/confs_sg/conf_neus.conf \
# --data_split_dir $data_path/$Scene/train/ \
# --data_split_dir_test $data_path/$Scene/test/ \
# --exps_folder_name $save_path \
# --expname 05_unknow_$Scene \
# --nepoch 2000 \
# --max_niter 200001 \
# --gamma 2.2 \
# --batch_size 1 \
# --roughness_warmup 5000 \
# --coordinate_type blender \
# --memory_capacity_level 15 \
# --secondary_batch_size 1024 \
# --secondary_train_interval 10 \
# --freeze_geometry \
# --geometry_neus $geometry_neus\
# 2>&1 | tee run_00.log