葛政(实习) commited on
Commit
22857d6
·
1 Parent(s): e3b4b18
Files changed (1) hide show
  1. 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
- "occumpy_mem",
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 occumpy_mem(cuda_device, mem_ratio=0.9):
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
  """