LASA / evaluation /dist_eval.sh
HaolinLiu's picture
first commit of codes and update readme.md
cc9780d
raw
history blame
550 Bytes
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