From c994963a841f7a491f53099154fdfd50e5035894 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 16 Jul 2019 18:59:46 +0200 Subject: [PATCH] updates --- cfg/yolov3-spp-1cls.cfg | 4 ++-- utils/torch_utils.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/cfg/yolov3-spp-1cls.cfg b/cfg/yolov3-spp-1cls.cfg index 9e989e91..a221589b 100644 --- a/cfg/yolov3-spp-1cls.cfg +++ b/cfg/yolov3-spp-1cls.cfg @@ -3,8 +3,8 @@ # batch=1 # subdivisions=1 # Training -batch=64 -subdivisions=32 +batch=16 +subdivisions=8 width=608 height=608 channels=3 diff --git a/utils/torch_utils.py b/utils/torch_utils.py index 5d08ba20..63d65d78 100644 --- a/utils/torch_utils.py +++ b/utils/torch_utils.py @@ -26,7 +26,7 @@ def select_device(force_cpu=False): c = 1024 ** 2 # bytes to MB ng = torch.cuda.device_count() x = [torch.cuda.get_device_properties(i) for i in range(ng)] - cuda_str = 'Using CUDA' + apex_str + cuda_str = 'Using CUDA ' + apex_str print("%sdevice0 _CudaDeviceProperties(name='%s', total_memory=%dMB)" % (cuda_str, x[0].name, x[0].total_memory / c)) if ng > 0: