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: