Use same metric for early stopping and saving best model

GateNLP / gate-lf-pytorch-json

PyTorch wrapper for the LearningFramework GATE plugin

Apache License 2.0

1 stars 2 forks source link

Use same metric for early stopping and saving best model #42

Closed johann-petrak closed 5 years ago

johann-petrak commented 5 years ago

The parameter --es_metric allows to choose accuracy or loss for the early stopping decision, but the best model is always saved based on the best validation accuracy. We should use the same metric there as for early stopping.