Closed sumedhvdatar closed 4 years ago
BIT model is lite weight but the model is taking entire GPU memory. Need a way to free up memory.
This is not nearly enough information. Some frameworks, for example TF and Jax, allocate all GPU memory by default. This is a framework-level setting and not related to our code release.
BIT model is lite weight but the model is taking entire GPU memory. Need a way to free up memory.