Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How long does it take to train? #3

Closed
peng666 opened this issue May 15, 2021 · 4 comments
Closed

How long does it take to train? #3

peng666 opened this issue May 15, 2021 · 4 comments

Comments

@peng666
Copy link

peng666 commented May 15, 2021

How long does it take to train? I train PCN about 3 epochs a day.

@paul007pl
Copy link
Owner

I remember that it should take around a few hours for training pcn (2048pts, 100 epochs, with CD loss).
I would like to suggest that you can discard the emd loss evaluation (very slow, especially for high resolution), which can help you save time.
Good Luck~

@peng666
Copy link
Author

peng666 commented May 18, 2021

But I've trained with CD as a loss function, still 3 epochs a day. I'm using a dense point cloud (16384 points). Is there any mistake here?

@paul007pl
Copy link
Owner

paul007pl commented May 18, 2021

I know you are using CD for training, but please check if you still evaluate the emd loss during training.
It is reasonable that you should take a longer time for training with 16384 points.
If I did not remember wrongly, it takes around 1 day for training PCN (16384 points, CD loss, 100 epochs) with a V100 GPU.
I suggest that you can try 2048 first. (I also did in this way...)
If your method works better than the others on 2048 pts, it should probably outperform the others on 16384 pts also.

@peng666
Copy link
Author

peng666 commented May 19, 2021

Your suggestion is very good. Thank you very much!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants