File size: 550 Bytes
cc9780d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
CUDA_VISIBLE_DEVICES='0,1,2,3' torchrun --master_port 15002 --nproc_per_node=2 \
evaluate_object_reconstruction.py \
--configs ../configs/finetune_triplane_diffusion.yaml \
--category arkit_chair arkit_stool \
--ae-pth ../output/ae/chair/best-checkpoint.pth \
--dm-pth ../output/finetune_dm/lowres_chair/best-checkpoint.pth \
--output_folder ../output_result/chair_result \
--data-pth ../data \
--eval_cd \
--reso 256 \
--save_mesh \
--save_par_points \
--save_image \
--save_surface
#check ./datasets/taxonomy to see how sub categories are defined
|