葛政(实习)
commited on
Commit
·
22857d6
1
Parent(s):
e3b4b18
fix typos
Browse files- yolox/utils/metric.py +2 -2
yolox/utils/metric.py
CHANGED
@@ -14,7 +14,7 @@ __all__ = [
|
|
14 |
"AverageMeter",
|
15 |
"MeterBuffer",
|
16 |
"get_total_and_free_memory_in_Mb",
|
17 |
-
"
|
18 |
"gpu_mem_usage",
|
19 |
]
|
20 |
|
@@ -28,7 +28,7 @@ def get_total_and_free_memory_in_Mb(cuda_device):
|
|
28 |
return int(total), int(used)
|
29 |
|
30 |
|
31 |
-
def
|
32 |
"""
|
33 |
pre-allocate gpu memory for training to avoid memory Fragmentation.
|
34 |
"""
|
|
|
14 |
"AverageMeter",
|
15 |
"MeterBuffer",
|
16 |
"get_total_and_free_memory_in_Mb",
|
17 |
+
"occupy_mem",
|
18 |
"gpu_mem_usage",
|
19 |
]
|
20 |
|
|
|
28 |
return int(total), int(used)
|
29 |
|
30 |
|
31 |
+
def occupy_mem(cuda_device, mem_ratio=0.9):
|
32 |
"""
|
33 |
pre-allocate gpu memory for training to avoid memory Fragmentation.
|
34 |
"""
|