Qualcomm-AI-research / geometric-algebra-transformer

BSD 3-Clause Clear License
154 stars 22 forks source link

Memory leak during training #2

Closed doloresgarcia closed 8 months ago

doloresgarcia commented 8 months ago

I use the gatr model with a pytorch lightning wrapper for the training and a custom dataset. Using this model is resulting in a host memory leak, while other models do not result in that problem. Have you observed this in your trainings? image