-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathuncond_command
27 lines (21 loc) · 4.67 KB
/
uncond_command
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
arr=("1129_730_800k" "1129_30_800k")
scp -P 16999 root@172.31.233.19:"/root/repo/python/outputs_brepnet_diffusion/1129_li_lr1e-5_730_4gpu/24-11-18-07-18-28/lightning_logs/version_0/checkpoints/last.ckpt" /mnt/d/img2brep/uncond_ckpt/${arr[1]}.ckpt
item=1111_adamw_1m
python -m src.brepnet.train_diffusion trainer.evaluate=true dataset.name=Dummy_dataset dataset.length=8000 trainer.batch_size=1000 trainer.num_worker=2 trainer.gpu=8 trainer.accelerator="32-true" model.name=Diffusion_condition model.autoencoder_weights=/mnt/d/img2brep/1119_deepcad_aug1_11k.ckpt model.autoencoder=AutoEncoder_1119_light model.with_intersection=true model.in_channels=6 model.dim_shape=768 model.dim_latent=8 model.gaussian_weights=1e-6 trainer.exp_name=test model.pad_method=random model.diffusion_latent=768 model.diffusion_type=epsilon model.gaussian_weights=1e-6 trainer.test_output_dir=/mnt/d/img2brep/uncond_results/1203/${item} trainer.resume_from_checkpoint=/mnt/d/img2brep/uncond_ckpt/${item}.ckpt model.num_max_faces=30 model.beta_schedule=linear model.addition_tag=false model.name=Diffusion_condition
python -m src.brepnet.train_diffusion trainer.evaluate=true dataset.name=Dummy_dataset dataset.length=2000 trainer.batch_size=250 trainer.num_worker=2 trainer.gpu=8 trainer.accelerator="32-true" model.name=Diffusion_condition model.autoencoder_weights=/mnt/d/img2brep/1119_deepcad_aug1_11k.ckpt model.autoencoder=AutoEncoder_1119_light model.with_intersection=true model.in_channels=6 model.dim_shape=768 model.dim_latent=8 model.gaussian_weights=1e-6 trainer.exp_name=test model.pad_method=random model.diffusion_latent=768 model.diffusion_type=epsilon model.gaussian_weights=1e-6 trainer.test_output_dir=/mnt/d/img2brep/uncond_results/1203/${item} trainer.resume_from_checkpoint=/mnt/d/img2brep/uncond_ckpt/${item}.ckpt model.num_max_faces=30 model.beta_schedule=linear model.addition_tag=false model.name=Diffusion_condition
item=/mnt/d/img2brep/uncond_results/1203/1111_adamw_1m
python -m src.brepnet.post.construct_brep --data_root ${item} --out_root ${item}_post --use_ray --num_cpus 160 --drop_num 3 --from_scratch && sleep 20
python -m src.brepnet.eval.check_valid --data_root ${item}_post
python -m src.brepnet.viz.sort_and_merge --data_root ${item}_post --out_root ${item}_choose --sort --valid --index --use_ray
python -m src.brepnet.eval.sample_points --data_root ${item}_post --out_root ${item}_pcd --valid
python -m src.brepnet.eval.eval_brepgen --real /mnt/d/img2brep/deepcad_test_pcd --fake ${item}_pcd
python -m src.brepnet.eval.eval_unique_novel --fake_root ${item} --fake_post ${item}_post --train_root /mnt/d/img2brep/brepgen_train --use_ray --min_face 0 --only_unique --atol 0.05
python -m src.brepnet.viz.find_nearest_pc_cd --fake_post ${item}_post --fake_pcd_root ${item}_pcd --train_root /mnt/d/img2brep/brepgen_train --txt ./src/brepnet/data/list/deduplicated_deepcad_training_7_30.txt --num_gpus 8
python -m src.brepnet.eval.eval_unique_novel --fake_root ${item} --fake_post ${item}_post --train_root /mnt/d/img2brep/brepgen_train --use_ray --txt ./src/brepnet/data/list/deduplicated_deepcad_training_7_30.txt --min_face 0 --n_bit 4
cd /root/repo/python/src/brepnet/eval/lfd/evaluation_scripts/compute_lfd_feat && python -m compute_lfd_feat_multiprocess --gen_path ${item}_post --save_path ${item}_lfd --prefix recon_brep.stl && cd /root/repo/python/src/brepnet/eval/lfd/evaluation_scripts && python -m compute_lfd --dataset_path /mnt/d/img2brep/brepgen_train_lfd_feat --gen_path ${item}_lfd --save_name ${item}_lfd.pkl --num_workers 8 --list /root/repo/python/src/brepnet/data/list/deduplicated_deepcad_training_7_30.txt && cd /root/repo/python/
python -m src.brepnet.eval.viz_lfd ${item}_lfd.pkl ${item}_lfd.png
# 30
python -m src.brepnet.viz.find_nearest_pc_cd --fake_post ${item}_post --fake_pcd_root ${item}_pcd --train_root /mnt/d/img2brep/brepgen_train --txt ./src/brepnet/data/list/deduplicated_deepcad_training_30.txt --num_gpus 8
python -m src.brepnet.eval.eval_unique_novel --fake_root ${item} --fake_post ${item}_post --train_root /mnt/d/img2brep/brepgen_train --use_ray --txt ./src/brepnet/data/list/deduplicated_deepcad_training_30.txt --min_face 0 --n_bit 4
cd /root/repo/python/src/brepnet/eval/lfd/evaluation_scripts/compute_lfd_feat && python -m compute_lfd_feat_multiprocess --gen_path ${item}_post --save_path ${item}_lfd --prefix recon_brep.stl && cd /root/repo/python/src/brepnet/eval/lfd/evaluation_scripts && python -m compute_lfd --dataset_path /mnt/d/img2brep/brepgen_train_lfd_feat --gen_path ${item}_lfd --save_name ${item}_lfd.pkl --num_workers 8 --list /root/repo/python/src/brepnet/data/list/deduplicated_deepcad_training_30.txt && cd /root/repo/python/
python -m src.brepnet.eval.viz_lfd ${item}_lfd.pkl ${item}_lfd.png