mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

minor

pull/173/head
KaiyangZhou 2019-04-19 11:52:02 +01:00
parent 1decc2473a
commit 4b93121496
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/engine/image/softmax.py
View File

@ -114,7 +114,7 @@ class ImageSoftmaxEngine(engine.Engine):
losses.update(loss.item(), pids.size(0)) losses.update(loss.item(), pids.size(0))
accs.update(metrics.accuracy(outputs, pids)[0].item()) accs.update(metrics.accuracy(outputs, pids)[0].item())
if (batch_idx+1) % print_freq==0: if (batch_idx+1) % print_freq == 0:
print('Epoch: [{0}][{1}/{2}]\t' print('Epoch: [{0}][{1}/{2}]\t'
'Time {batch_time.val:.3f} ({batch_time.avg:.3f})\t' 'Time {batch_time.val:.3f} ({batch_time.avg:.3f})\t'
'Data {data_time.val:.3f} ({data_time.avg:.3f})\t' 'Data {data_time.val:.3f} ({data_time.avg:.3f})\t'

2
torchreid/engine/image/triplet.py
View File

@ -129,7 +129,7 @@ class ImageTripletEngine(engine.Engine):
losses_x.update(loss_x.item(), pids.size(0)) losses_x.update(loss_x.item(), pids.size(0))
accs.update(metrics.accuracy(outputs, pids)[0].item()) accs.update(metrics.accuracy(outputs, pids)[0].item())
if (batch_idx+1) % print_freq==0: if (batch_idx+1) % print_freq == 0:
print('Epoch: [{0}][{1}/{2}]\t' print('Epoch: [{0}][{1}/{2}]\t'
'Time {batch_time.val:.3f} ({batch_time.avg:.3f})\t' 'Time {batch_time.val:.3f} ({batch_time.avg:.3f})\t'
'Data {data_time.val:.3f} ({data_time.avg:.3f})\t' 'Data {data_time.val:.3f} ({data_time.avg:.3f})\t'