add time counter to evaluate mode
parent
fd838eaaa0
commit
cb0cfa3e19
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue