Skip to content

GitLab

Unverified Commit 4c5bd921 authored by William Falcon's avatar William Falcon Committed by GitHub
Browse files

Update run_pl_glue.py (#4117)

parent 5282b31d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/glue/run_pl_glue.py
View file @ 4c5bd921
...@@ -124,7 +124,7 @@ class GLUETransformer(BaseTransformer): ...@@ -124,7 +124,7 @@ class GLUETransformer(BaseTransformer):
ret["log"] = results ret["log"] = results
return ret, preds_list, out_label_list return ret, preds_list, out_label_list
def validation_end(self, outputs: list) -> dict: def validation_epoch_end(self, outputs: list) -> dict:
ret, preds, targets = self._eval_end(outputs) ret, preds, targets = self._eval_end(outputs)
logs = ret["log"] logs = ret["log"]
return {"val_loss": logs["val_loss"], "log": logs, "progress_bar": logs} return {"val_loss": logs["val_loss"], "log": logs, "progress_bar": logs}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment