File size: 331 Bytes
cc9780d |
1 2 3 4 5 6 7 |
CUDA_VISIBLE_DEVICES='0,1,2,3' torchrun --master_port 15000 --nproc_per_node=2 \
extract_img_vit_features.py \
--batch_size 24 \
--ckpt_path ../data/open_clip_pytorch_model.bin \
--category arkit_chair 03001627 future_chair arkit_stool future_stool ABO_chair #sub category
#--category 02871439 future_shelf ABO_shelf arkit_shelf \
|