Browse Source

Compute loss on final val (#5017)

modifyDataloader
Glenn Jocher GitHub 3 years ago
parent
commit
89a6e9ffa7
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      train.py

+ 2
- 1
train.py View File

@@ -419,7 +419,8 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
save_json=is_coco,
verbose=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)
LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")

Loading…
Cancel
Save