From 3a338f12b0548d39282673617c41acf6b48ea40d Mon Sep 17 00:00:00 2001 From: matteogu Date: Fri, 24 May 2024 18:29:13 -0700 Subject: [PATCH 1/2] compatibility fix kornia --- examples/unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/unet.py b/examples/unet.py index ef74947..4edffa0 100644 --- a/examples/unet.py +++ b/examples/unet.py @@ -147,7 +147,7 @@ def __init__(self): self.register_buffer('f', f) def forward(self, x): - return kornia.filter2D(x, self.f, normalized=True) + return kornia.filters.filter2d(x, self.f, normalized=True) class Downsample(nn.Module): From 128a6a6d75f37a6ed20c92e4d1418d3589d837cf Mon Sep 17 00:00:00 2001 From: matteogu Date: Fri, 24 May 2024 19:33:57 -0700 Subject: [PATCH 2/2] visualization of lorenz bug --- examples/latent_sde_lorenz.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/latent_sde_lorenz.py b/examples/latent_sde_lorenz.py index e4dd339..5eb1d8c 100644 --- a/examples/latent_sde_lorenz.py +++ b/examples/latent_sde_lorenz.py @@ -238,7 +238,7 @@ def vis(xs, ts, latent_sde, bm_vis, img_path, num_samples=10): # Left plot: data. z1, z2, z3 = np.split(xs.cpu().numpy(), indices_or_sections=3, axis=-1) [ax00.plot(z1[:, i, 0], z2[:, i, 0], z3[:, i, 0]) for i in range(num_samples)] - ax00.scatter(z1[0, :num_samples, 0], z2[0, :num_samples, 0], z3[0, :10, 0], marker='x') + ax00.scatter(z1[0, :num_samples, 0], z2[0, :num_samples, 0], z3[0, :num_samples, 0], marker='x') ax00.set_yticklabels([]) ax00.set_xticklabels([]) ax00.set_zticklabels([]) @@ -255,7 +255,7 @@ def vis(xs, ts, latent_sde, bm_vis, img_path, num_samples=10): z1, z2, z3 = np.split(xs, indices_or_sections=3, axis=-1) [ax01.plot(z1[:, i, 0], z2[:, i, 0], z3[:, i, 0]) for i in range(num_samples)] - ax01.scatter(z1[0, :num_samples, 0], z2[0, :num_samples, 0], z3[0, :10, 0], marker='x') + ax01.scatter(z1[0, :num_samples, 0], z2[0, :num_samples, 0], z3[0, :num_samples, 0], marker='x') ax01.set_yticklabels([]) ax01.set_xticklabels([]) ax01.set_zticklabels([])