diff --git a/examples/backends/plot_Sinkhorn_gradients.py b/examples/backends/plot_Sinkhorn_gradients.py index e5e8f1909..229a952b1 100644 --- a/examples/backends/plot_Sinkhorn_gradients.py +++ b/examples/backends/plot_Sinkhorn_gradients.py @@ -1,4 +1,3 @@ -# %% # -*- coding: utf-8 -*- """ ================================================ @@ -27,7 +26,7 @@ # %% parameters -n_trials = 30 +n_trials = 10 times_autodiff = torch.zeros(n_trials) times_envelope = torch.zeros(n_trials) times_last_step = torch.zeros(n_trials)