xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix print val map

Browse Source
This commit is contained in:
wa22 2024-05-26 03:49:16 +00:00
parent 53b713413b
commit 7a8f9643d2
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
ultralytics/engine/validator.py
View File

@ -196,8 +196,7 @@ class BaseValidator:
self.check_stats(stats) self.check_stats(stats)
self.speed = dict(zip(self.speed.keys(), (x.t / len(self.dataloader.dataset) * 1e3 for x in dt))) self.speed = dict(zip(self.speed.keys(), (x.t / len(self.dataloader.dataset) * 1e3 for x in dt)))
self.finalize_metrics() self.finalize_metrics()
if not (self.args.save_json and self.jdict): self.print_results()
self.print_results()
self.run_callbacks("on_val_end") self.run_callbacks("on_val_end")
if self.training: if self.training:
model.float() model.float()