hansen7 / OcCo

[ICCV '21] "Unsupervised Point Cloud Pre-training via Occlusion Completion"
https://hansen7.github.io/OcCo/
MIT License
233 stars 26 forks source link

Experiment setting of fine-tuning on ModelNet40 #22

Closed SimingYan closed 2 years ago

SimingYan commented 2 years ago

Hi Hansen,

Thanks for your great work. I have one quick question here.

May I ask what's the detailed experiment setting of fine-tuning on ModelNet40? Could you please provide the command of this task?

Now I'm using:

python train_cls.py --model=dgcnn_cls --dataset=modelnet40 --log_dir=dgcnn_occo --num_point=2048 --restore --restore_path=./path/to/dgcnn_occ_cls.pth

If I made any mistakes, please let me know! Thank you!

hansen7 commented 2 years ago

Hi @SimingYan . Thanks for your interest, it is the same setting as we trained on ScanObjectNN (in readme file)

SimingYan commented 2 years ago

Got it. Thanks for the clarification.