add time counter to evaluate mode

pull/17/head
KaiyangZhou 2018-03-11 21:25:26 +00:00
parent fd838eaaa0
commit cb0cfa3e19
1 changed files with 5 additions and 1 deletions

View File

@ -123,7 +123,11 @@ def main():
if args.evaluate:
print("Evaluate only")
test()
start_time = time.time()
test(model, queryloader, galleryloader, use_gpu)
elapsed = time.time() - start_time
elapsed = str(datetime.timedelta(seconds=elapsed))
print("Finished.\t Total elapsed time: {}".format(elapsed))
return
start_time = time.time()