mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Compute loss on final val (#5017)

Browse Source
This commit is contained in:
Glenn Jocher 2021-09-30 15:16:11 -07:00 committed by GitHub
parent b20e381492
commit 89a6e9ffa7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
train.py
View File

@ -419,7 +419,8 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
save_json=is_coco, save_json=is_coco,
verbose=True, verbose=True,
plots=True, plots=True,
callbacks=callbacks) # val best model with plots callbacks=callbacks,
compute_loss=compute_loss) # val best model with plots
callbacks.run('on_train_end', last, best, plots, epoch) callbacks.run('on_train_end', last, best, plots, epoch)
LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}") LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")