Skip to content

Commit

Permalink
add flushing of val epoch resluts
Browse files Browse the repository at this point in the history
  • Loading branch information
AUdaltsova authored Sep 25, 2024
1 parent b5fa2d9 commit e8292d9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pvnet/models/base_model.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@
"""Base model for all PVNet submodels"""
import json
import logging
import os
Expand Down Expand Up @@ -743,6 +742,7 @@ def on_validation_epoch_end(self):
print("Failed to log validation results to wandb")
print(e)

self.validation_epoch_results = []
horizon_maes_dict = self._horizon_maes.flush()

# Create the horizon accuracy curve
Expand Down

0 comments on commit e8292d9

Please sign in to comment.