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([]) 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):