number of epochs to 10k

mpezeshki · Jan 10, 2021 · bc5c296 · bc5c296
1 parent 51d131a
commit bc5c296
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Figure_3/moon_fig_3.py b/Figure_3/moon_fig_3.py
@@ -18,7 +18,7 @@
 # A two-layer neural netowrk with the following number of hidden units
 hidden_dim = 500
 n_samples = 300
-epochs = 1000
+epochs = 10000
 
 dataset = lambda seed: two_moons_dataset(seed, margin=+0.1)
 experiments = [