diff options
Diffstat (limited to 'examples/backends/plot_stoch_continuous_ot_pytorch.py')
-rw-r--r-- | examples/backends/plot_stoch_continuous_ot_pytorch.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/backends/plot_stoch_continuous_ot_pytorch.py b/examples/backends/plot_stoch_continuous_ot_pytorch.py index 714a5d3..e642986 100644 --- a/examples/backends/plot_stoch_continuous_ot_pytorch.py +++ b/examples/backends/plot_stoch_continuous_ot_pytorch.py @@ -27,8 +27,8 @@ import ot.plot torch.manual_seed(42) np.random.seed(42) -n_source_samples = 10000 -n_target_samples = 10000 +n_source_samples = 1000 +n_target_samples = 1000 theta = 2 * np.pi / 20 noise_level = 0.1 @@ -89,7 +89,7 @@ reg = 1 optimizer = torch.optim.Adam(list(u.parameters()) + list(v.parameters()), lr=.005) # number of iteration -n_iter = 1000 +n_iter = 500 n_batch = 500 |