Update run_pl_ner.py (#4118)

5282b31d · William Falcon · GitHub · 1e616c0a · 5282b31d
Unverified Commit 5282b31d authored May 02, 2020 by William Falcon Committed by GitHub May 02, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

examples/ner/run_pl_ner.py examples/ner/run_pl_ner.py +1 -2

No files found.
--- a/examples/ner/run_pl_ner.py
+++ b/examples/ner/run_pl_ner.py
@@ -130,8 +130,7 @@ class NERTransformer(BaseTransformer):
        ret["log"] = results
        return ret, preds_list, out_label_list

-    def validation_end(self, outputs):
-        # todo: update to validation_epoch_end instead of deprecated validation_end
+    def validation_epoch_end(self, outputs):
        # when stable
        ret, preds, targets = self._eval_end(outputs)
        logs = ret["log"]