SuyeonJ commited on
Commit
54e8a63
·
verified ·
1 Parent(s): dcf0f2f

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. utils/misc.py +2 -2
utils/misc.py CHANGED
@@ -37,9 +37,9 @@ def print_cuda_statistics():
37
  logger.info('__Devices')
38
  call(["nvidia-smi", "--format=csv",
39
  "--query-gpu=index,name,driver_version,memory.total,memory.used,memory.free"])
40
- logger.info('Active CUDA Device: GPU {}'.format("cpu")#torch.cuda.current_device()))
41
  logger.info('Available devices {}'.format(torch.cuda.device_count()))
42
- logger.info('Current cuda device {}'.format("cpu")#torch.cuda.current_device()))
43
 
44
 
45
  class SmoothedValue(object):
 
37
  logger.info('__Devices')
38
  call(["nvidia-smi", "--format=csv",
39
  "--query-gpu=index,name,driver_version,memory.total,memory.used,memory.free"])
40
+ #logger.info('Active CUDA Device: GPU {}'.format("cpu")#torch.cuda.current_device()))
41
  logger.info('Available devices {}'.format(torch.cuda.device_count()))
42
+ #logger.info('Current cuda device {}'.format("cpu")#torch.cuda.current_device()))
43
 
44
 
45
  class SmoothedValue(object):