issues
search
openai
/
weak-to-strong
MIT License
2.5k
stars
305
forks
source link
Enhance clear_mem function readability and precision
#7
Closed
Nagi-ovo
closed
10 months ago
Nagi-ovo
commented
10 months ago
Purpose of this contribution☺️:
Improved docstring for clarity.(Refer to the
TORCH.CUDA.MEMORY_ALLOCATED
in
the PyTorch doc
.)
Switched to f-string for memory usage output.
Enhanced memory allocation precision with accurate GB calculation and two decimal formatting.
Purpose of this contribution☺️:
TORCH.CUDA.MEMORY_ALLOCATED
in the PyTorch doc.)