fizyr / keras-retinanet

Keras implementation of RetinaNet object detection.
Apache License 2.0
4.38k stars 1.96k forks source link

Enable earlystopping only if validation set is provided #1388

Closed ppizarror closed 4 years ago

ppizarror commented 4 years ago

Related issue: #1366 EarlyStopping class uses mAP by default, but mAP only is computet if evaluation dataset is provided. If one tries to train the model without any validation dataset, a warning appears:

RuntimeWarning: Early stopping conditioned on metric mAP which is not available. Available metrics are: loss,regression_loss,classification_loss,lr (self.monitor, ','.join(list(logs.keys()))), RuntimeWarning

hgaiser commented 4 years ago

Awesome, thank you :)