Open ememos opened 3 months ago
We need a feature that prints the size of the GPU memory during learning.
This feature added and activated when Optimus_p( ... , display_mem=True)
We need a feature that prints the size of the GPU memory during learning.