ultralytics 8.0.173 fix missing val plots and new Conda Guide (#4783)
This commit is contained in:
parent
602022a56e
commit
dfe6dfb1d2
8 changed files with 188 additions and 23 deletions
|
|
@ -571,6 +571,7 @@ class BaseTrainer:
|
|||
strip_optimizer(f) # strip optimizers
|
||||
if f is self.best:
|
||||
LOGGER.info(f'\nValidating {f}...')
|
||||
self.validator.args.plots = self.args.plots
|
||||
self.metrics = self.validator(model=f)
|
||||
self.metrics.pop('fitness', None)
|
||||
self.run_callbacks('on_fit_epoch_end')
|
||||
|
|
|
|||
|
|
@ -102,8 +102,8 @@ class BaseValidator:
|
|||
@smart_inference_mode()
|
||||
def __call__(self, trainer=None, model=None):
|
||||
"""
|
||||
Supports validation of a pre-trained model if passed or a model being trained
|
||||
if trainer is passed (trainer gets priority).
|
||||
Supports validation of a pre-trained model if passed or a model being trained if trainer is passed (trainer
|
||||
gets priority).
|
||||
"""
|
||||
self.training = trainer is not None
|
||||
augment = self.args.augment and (not self.training)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue