search

ARISE-Initiative / robomimic

robomimic: A Modular Framework for Robot Learning from Demonstration
MIT License
595 stars 182 forks source link

GPU #120

Closed Nimingez closed 9 months ago

Nimingez commented 9 months ago

My GPU is 3080Ti 。 My GPU memory usage is only 1527MB. What modifications can be made to speed up training

amandlek commented 9 months ago

I'm a little unclear on your question - but perhaps you could try increasing the batch size?