-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathcond_command
More file actions
34 lines (25 loc) · 5.35 KB
/
cond_command
File metadata and controls
34 lines (25 loc) · 5.35 KB
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
cache_str="trainer.evaluate=true dataset.name=Diffusion_dataset dataset.test_dataset=/mnt/d/deepcad_test_v6 trainer.batch_size=64 trainer.num_worker=4 trainer.gpu=1 trainer.accelerator="bf16-mixed" trainer.check_val_every_n_epoch=1 model.name=Diffusion_condition model.autoencoder_weights=/mnt/d/0925_gaussian_7m.ckpt model.autoencoder=AutoEncoder_0925 model.dim_shape=768 model.dim_latent=8 model.gaussian_weights=1e-6 trainer.exp_name=test dataset.deduplicate_list=1 dataset.pad_method=random model.pad_method=random model.diffusion_latent=768 model.num_max_faces=30 dataset.cached_condition=false dataset.num_max_faces=30"
# arr=(1111_730_li_mvr 1111_730_li_pc 1111_730_li_sketch 1111_730_li_svr 1111_730_sq_mvr 1111_730_sq_pc 1111_730_sq_sketch 1111_730_sq_svr)
scp -P 17999 root@172.31.233.126:"/root/repo/python/outputs_brepnet_diffusion/pc_gaussian_30/24-10-22-15-55-58/lightning_logs/version_0/checkpoints/last.ckpt" ${arr[1]}.ckpt
conds=("multi_img" "pc" "sketch" "single_img")
cond=sketch
item=1111_730_li_sketch
scheduler=linear
python -m src.brepnet.train_diffusion trainer.evaluate=true dataset.name=Diffusion_dataset dataset.face_z=/mnt/d/deepcad_730_v0 dataset.test_dataset=src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt trainer.batch_size=256 trainer.num_worker=0 trainer.gpu=8 trainer.accelerator="bf16-mixed" trainer.check_val_every_n_epoch=1 model.name=Diffusion_condition model.autoencoder=AutoEncoder_0925 model.dim_shape=768 model.dim_latent=8 model.gaussian_weights=1e-6 trainer.exp_name=test dataset.pad_method=random model.pad_method=random model.diffusion_latent=768 model.num_max_faces=30 dataset.cached_condition=false dataset.num_max_faces=30 model.diffusion_type=epsilon model.beta_schedule=${scheduler} model.gaussian_weights=1e-6 model.condition=${cond} dataset.condition=${cond} trainer.test_output_dir=/mnt/d/cond_results/${item} trainer.resume_from_checkpoint=/mnt/d/cond_ckpt/${item}.ckpt
item=1111_730_li_mvr
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/${item} --out_root /mnt/d/cond_results/${item}_post --num_cpus 160 --use_ray --drop_num 3
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_li_svr --out_root /mnt/d/cond_results/1111_730_li_svr_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_li_pc --out_root /mnt/d/cond_results/1111_730_li_pc_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_li_sketch --out_root /mnt/d/cond_results/1111_730_li_sketch_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_sq_mvr --out_root /mnt/d/cond_results/1111_730_sq_mvr_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_sq_svr --out_root /mnt/d/cond_results/1111_730_sq_svr_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_sq_pc --out_root /mnt/d/cond_results/1111_730_sq_pc_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.post.construct_brep --data_root /mnt/d/cond_results/1111_730_sq_sketch --out_root /mnt/d/cond_results/1111_730_sq_sketch_post2 --use_ray --num_cpus 160 --drop_num 0 --from_scratch
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_li_mvr_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_li_svr_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_li_pc_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_li_sketch_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_sq_mvr_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_sq_svr_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_sq_pc_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160
python -m src.brepnet.eval.eval_condition --eval_root /mnt/d/cond_results/1111_730_sq_sketch_post2 --gt_root /mnt/d/yilin/img2brep/deepcad_v6 --list src/brepnet/data/list/deduplicated_deepcad_testing_7_30.txt --use_ray --from_scratch --num_cpus 160