search

LMD0311 / PointMamba

[NeurIPS 2024] PointMamba: A Simple State Space Model for Point Cloud Analysis
https://arxiv.org/abs/2402.10739
Apache License 2.0
361 stars 24 forks source link

Testing #11

Closed hithqd closed 6 months ago

hithqd commented 6 months ago

Thank you for your great work. I have two questions.

(1) I would like to ask what is the startup script and code for the test ? Is that so？ CUDA_VISIBLE_DEVICES=<GPU> python main.py --test --config cfgs/finetune_modelnet.yaml --ckpts <path/to/model> --exp_name <name>

(2) The second question is about pre-training.

image After the pre-training is completed, several models will be saved. Which one is the best? Is the one named ckpt-last.pth the best?

LMD0311 commented 6 months ago

(1) Yes.

(2) Yes, we default to selecting the last checkpoint.