search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.82k stars 178 forks source link

imagenet training increasing gpu memory usage #128

Closed jeannotes closed 2 years ago

jeannotes commented 2 years ago

hi, recently I train imagenet using gpus, everything works fine, but as epochs increase, the gpu memory increase too, why?

GuillaumeLeclerc commented 2 years ago

Probably an issue in your code.

(If you want us to be able to help you we will need more than a single sentence)