Update utils.py (#82)

get_gpu_memory should return after the loop is over
This commit is contained in:
magic.chen 2023-05-23 16:02:13 +08:00 committed by GitHub
commit 528c0e78f6