Validate best.pt on train end (#4889)

* Validate best.pt on train end

* 0.7 iou for COCO only

* pass callbacks

* active model.float() if not half

* print Validating best.pt...

* add newline
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-26 11:54:53 -07:00
1 parent e687873436
commit d856c48298
2 files changed
+14 -16

No files matched your search

+1 -2
View File
@@ -133,8 +133,7 @@ def run(data,
# Half
half &= device.type != 'cpu' # half precision only supported on CUDA
if half:
model.half()
model.half() if half else model.float()
# Configure
model.eval()