2020-1-capstone-design2 / 2016104167

Go to a project
Toggle navigation Toggle navigation pinning
조현아
75bcfcee 1 parent aa191bf0

get GPU name

Inline Side-by-side
Showing 1 changed file with 2 additions and 2 deletions
...@@ -64,8 +64,8 @@ def train_child(args, model, dataset, subset_indx, device=None): ...@@ -64,8 +64,8 @@ def train_child(args, model, dataset, subset_indx, device=None):
64 _train_res = train_step(args, model, optimizer, scheduler, criterion, batch, step, None, device) 64 _train_res = train_step(args, model, optimizer, scheduler, criterion, batch, step, None, device)
65 65
66 if step % args.print_step == 0: 66 if step % args.print_step == 0:
67 - print('\n[+] Training step: {}/{}\tElapsed time: {:.2f}min\tLearning rate: {}\tDevice: {}'.format( 67 + print('\n[+] Training step: {}/{}\tElapsed time: {:.2f}min\tLearning rate: {}\tDevice name: {}'.format(
68 - step, args.max_step,(time.time()-start_t)/60, optimizer.param_groups[0]['lr'], device)) 68 + step, args.max_step,(time.time()-start_t)/60, optimizer.param_groups[0]['lr'], torch.cuda.get_device_name(0)))
69 69
70 print(' Acc@1 : {:.3f}%'.format(_train_res[0].data.cpu().numpy()[0]*100)) 70 print(' Acc@1 : {:.3f}%'.format(_train_res[0].data.cpu().numpy()[0]*100))
71 print(' Acc@5 : {:.3f}%'.format(_train_res[1].data.cpu().numpy()[0]*100)) 71 print(' Acc@5 : {:.3f}%'.format(_train_res[1].data.cpu().numpy()[0]*100))
......