Closed thomashopkins32 closed 1 year ago
This refactors the training script, closing #9, and creates utilities to inspect CUDA memory usage throughout the training loop for both full and mixed precision training.
Closes #9 Closes #2
This refactors the training script, closing #9, and creates utilities to inspect CUDA memory usage throughout the training loop for both full and mixed precision training.
Closes #9 Closes #2