diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py
index 15548ff..717025e 100644
--- a/app/yolov5/train_server.py
+++ b/app/yolov5/train_server.py
@@ -609,7 +609,7 @@ def main(opt,data_list,id,callbacks=Callbacks()):
 
     # DDP mode
     device = select_device(opt.device, batch_size=opt.batch_size)
-    print(device.type)
+    print('\n','device:',device,'\n')
     # if LOCAL_RANK != -1:
     #     msg = 'is not compatible with YOLOv5 Multi-GPU DDP training'
     #     assert not opt.image_weights, f'--image-weights {msg}'