Upload folder using huggingface_hub
Browse files- 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):
|