Fix torch cache cleanup issues

rapidsai / crossfit

Metric calculation library

Apache License 2.0

2 stars 6 forks source link

Closed VibhuJawa closed 2 months ago

VibhuJawa commented 2 months ago

This PR fixes the memory issues caused b/w Pytorch and Cache clean up.