search

jsll / pytorch_6dof-graspnet

MIT License
146 stars 55 forks source link

Training plots #12

Closed richardrl closed 3 years ago

richardrl commented 3 years ago

Hi, do you still have the training plots? How long does it take in terms of wall clock time / what GPU hardware?

jsll commented 3 years ago

The training plots should be in the pre-trained model folder which I recently fixed.

As far as I can remember, you could get decent results after about 5 hours if you train the grasp sampler with implicit Maximum Likelihood Estimation. If you use the regular VAE loss it will take longer.