MinkaiXu/GeoLDM

Training time for QM9 dataset

sudarshan1994 opened this issue · 2 comments

How long does it take to train the model on qm9 for 3000 epochs with batch size of 64? On my machine it seems like even one epoch would take 5 hours with a batch size of 64. Are the hyper parameters I am using correct?

Thank you for your work!!!

I fixed some package issues and now it is taking me 500 seconds per epoch, is this standard?