transformer_lens.utilities.gpu_utils module¶
gpu_utils.
This module contains varied utility functions related to GPUs.
- transformer_lens.utilities.gpu_utils.print_gpu_mem(step_name='')¶
Back to top