Skip to content
This repository was archived by the owner on Jan 29, 2024. It is now read-only.

Commit 5b31822

Browse files
committed
Small fix.
1 parent f302b03 commit 5b31822

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trainers/trainer.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -92,4 +92,4 @@ def run(self, start_epoch, total_epoch, train_loader, query_loader, gallery_load
9292
torch.save(self.model.state_dict(), os.path.join(out_dir, save_filename))
9393
self.logger.info(save_filename + ' saved.\n')
9494

95-
print('Best mAP {:.1%}, achieved at Epoch [{}]'.format(best_mAP, best_epoch))
95+
self.logger.info('Best mAP {:.1%}, achieved at Epoch [{}]'.format(best_mAP, best_epoch))

0 commit comments

Comments
 (0)