search

ai-computing / aicomp

Other
13 stars 1 forks source link

Support of GPU Memory Size Print During Learning #11

Closed ememos closed 4 days ago

ememos commented 6 months ago

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

ememos commented 6 months ago

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