search

zubair-irshad / CenterSnap

Pytorch code for ICRA'22 paper: "Single-Shot Multi-Object 3D Shape Reconstruction and Categorical 6D Pose and Size Estimation"
https://zubair-irshad.github.io/projects/CenterSnap.html
286 stars 47 forks source link

Training time #6

Closed WenM1222 closed 2 years ago

WenM1222 commented 2 years ago

Thanks for sharing your great works! May I know how much time the training took?

zubair-irshad commented 2 years ago

Training time for synthetic data is around 2 days for 25 epochs. For real finetuning, the training should run for a couple hours for a few epochs (1-5)

zubair-irshad commented 2 years ago

Closing due to inactivity!

DavidYaonanZhu commented 1 year ago

Hi, @zubair-irshad Thanks for amazing work,

May I ask the detail about your hardware? How many GPUs did you use to train? And how many GPU memories?

I guess the dataset is so large 800GB... it will take more time with only one GPU.