def validation_epoch_end(self, outputs: List) -> None:
result_predictions: List[dict] = []
result_gt: List[dict] = []
for output in outputs:
result_predictions += output["predictions"]
result_gt += output["gt"]
_, _, average_precision = recall_precision(result_gt, result_predictions, 0.5)
self.log("epoch", self.trainer.current_epoch, on_step=False, on_epoch=True, logger=True)
self.log("val_loss", average_precision, on_step=False, on_epoch=True, logger=True)
def validation_epoch_end(self, outputs: List) -> Dict[str, Any]:
result_predictions: List[dict] = []
result_gt: List[dict] = []
for output in outputs:
result_predictions += output["predictions"]
result_gt += output["gt"]
_, _, average_precision = recall_precision(result_gt, result_predictions, 0.5)
logs = {"epoch": self.trainer.current_epoch, "[email protected]": average_precision}
return {"val_loss": average_precision, "log": logs}