Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Remove unused variable
Browse files Browse the repository at this point in the history
gbruno16 committed Jul 22, 2024
1 parent 6d366c6 commit 99e9d6f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions graph_weather/models/gencast/train.py
Original file line number Diff line number Diff line change
@@ -159,7 +159,7 @@ def plot_sample(self, prev_inputs, target_residuals):
preds = sampler.sample(self.model, prev_inputs)

fig1, ax = plt.subplots(2)
im = ax[0].imshow(preds[0, :, :, 78].T.cpu(), origin="lower", cmap="RdBu", vmin=-5, vmax=5)
ax[0].imshow(preds[0, :, :, 78].T.cpu(), origin="lower", cmap="RdBu", vmin=-5, vmax=5)
ax[0].set_xticks([])
ax[0].set_yticks([])
ax[0].set_title("Diffusion sampling prediction")
@@ -170,7 +170,7 @@ def plot_sample(self, prev_inputs, target_residuals):
ax[1].set_title("Ground truth")

fig2, ax = plt.subplots(2)
im = ax[0].imshow(preds[0, :, :, 12].T.cpu(), origin="lower", cmap="RdBu", vmin=-5, vmax=5)
ax[0].imshow(preds[0, :, :, 12].T.cpu(), origin="lower", cmap="RdBu", vmin=-5, vmax=5)
ax[0].set_xticks([])
ax[0].set_yticks([])
ax[0].set_title("Diffusion sampling prediction")

0 comments on commit 99e9d6f

Please sign in to comment.