diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py index 758dc65..4ffe49c 100644 --- a/pvnet_summation/models/base_model.py +++ b/pvnet_summation/models/base_model.py @@ -16,7 +16,6 @@ from pvnet_summation.utils import plot_forecasts - logger = logging.getLogger(__name__) activities = [torch.profiler.ProfilerActivity.CPU] @@ -52,7 +51,6 @@ def __init__( ) self.pvnet_model.requires_grad_(False) - self._optimizer = optimizer # Model must have lr to allow tuning diff --git a/pvnet_summation/models/model.py b/pvnet_summation/models/model.py index 0c9c953..5256be8 100644 --- a/pvnet_summation/models/model.py +++ b/pvnet_summation/models/model.py @@ -110,4 +110,4 @@ def forward(self, x): out = F.leaky_relu(gsp_sum + out) - return out \ No newline at end of file + return out diff --git a/pvnet_summation/utils.py b/pvnet_summation/utils.py index 91214de..9f7d179 100644 --- a/pvnet_summation/utils.py +++ b/pvnet_summation/utils.py @@ -24,7 +24,7 @@ def plot_forecasts(y, y_hat, times, batch_idx=None, quantiles=None, y_sum=None): continue ax.plot(times_utc[i], y[i], marker=".", color="k", label=r"$y$") - + if y_sum is not None: ax.plot( times_utc[i], y_sum[i], marker=".", linestyle="--", color="k", label=r"$y_{sum}$"