get GPU name

@@ -64,8 +64,8 @@ def train_child(args, model, dataset, subset_indx, device=None):
         _train_res = train_step(args, model, optimizer, scheduler, criterion, batch, step, None, device)
 
         if step % args.print_step == 0:
-            print('\n[+] Training step: {}/{}\tElapsed time: {:.2f}min\tLearning rate: {}\tDevice: {}'.format(
-                step, args.max_step,(time.time()-start_t)/60, optimizer.param_groups[0]['lr'], device))
+            print('\n[+] Training step: {}/{}\tElapsed time: {:.2f}min\tLearning rate: {}\tDevice name: {}'.format(
+                step, args.max_step,(time.time()-start_t)/60, optimizer.param_groups[0]['lr'], torch.cuda.get_device_name(0)))
 
             print('  Acc@1 : {:.3f}%'.format(_train_res[0].data.cpu().numpy()[0]*100))
             print('  Acc@5 : {:.3f}%'.format(_train_res[1].data.cpu().numpy()[0]*100))