"test/git@developer.sourcefind.cn:hehl2/torchaudio.git" did not exist on "f0bc00c980012badea8db011f84a0e9ef33ba6c1"
Unverified Commit 4c5bd921 authored by William Falcon's avatar William Falcon Committed by GitHub
Browse files

Update run_pl_glue.py (#4117)

parent 5282b31d
......@@ -124,7 +124,7 @@ class GLUETransformer(BaseTransformer):
ret["log"] = results
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)
logs = ret["log"]
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