search

MohamedAfham / CrossPoint

Official implementation of "CrossPoint: Self-Supervised Cross-Modal Contrastive Learning for 3D Point Cloud Understanding" (CVPR, 2022)
https://mohamedafham.github.io/CrossPoint/
228 stars 29 forks source link

distributed training for CrossPoint #11

Open auniquesun opened 2 years ago

auniquesun commented 2 years ago

@MohamedAfham I have succefully integrated the PyTorch DistributedDataParallel mechanism into your codebase, which accelerates the training procedure remarkbly and achieves a similar performance with the paper reported.

Later on I want to pull a request to your repository, thank you.

MohamedAfham commented 2 years ago

Thank you for performing Distributed parallel mechanism and verifying the reported results.

I would love to merge the code once you do the pull request.

auniquesun commented 2 years ago

@MohamedAfham Since the pull request has not been merged, let me know if you have any concern about this pull request , thank you.

MohamedAfham commented 2 years ago

As mentioned here #13 , I'll review the PR as soon as I'm available and get back to you. Sorry for the inconvenience.

auniquesun commented 1 year ago

The distributed implementation of CrossPoint training and inference is available here !