search

ai-computing / aicomp

Other
6 stars 0 forks source link

Support of GPU Memory Size Print During Learning #11

Open ememos opened 3 months ago

ememos commented 3 months ago

We need a feature that prints the size of the GPU memory during learning.

ememos commented 2 months ago

This feature added and activated when Optimus_p( ... , display_mem=True)