koila
Prevent PyTorch's CUDA error: out of memory in just 1 line of code.
Archived
2k stars
11 watching
63 forks
Language: Python
last commit: about 1 month ago deep-learninggradient-accumulationlazy-evaluationmachine-learningmemory-managementneural-networkout-of-memorypythonpytorch