search

snuailab / waffle_hub

🧇Waffle Hub🧇 is Deep Learning 🛠️Open Source Framework Adapter. It provides you 😎no-code, 😏easy training and inference!
GNU General Public License v3.0
40 stars 13 forks source link

Hotfix: Holding memory after `hold = False` operation #270

Closed Jiyong-Jeon closed 1 year ago

Jiyong-Jeon commented 1 year ago

I modified to unallocate GPU memory after the hold = False operation(train, evaluate, export, inference) except for the primary memory allocated by pytorch.

We will move the "hold" function to "waffle_store", so we will end it as a temporary measure for now. Nevertheless, if you have a critical problem, please write a new issue. resolve: #265