changed schedule peak_value to 1e-2 for the svgp notebook so it does …

…nott give nans (#324)
JaxGaussianProcesses · Jun 22, 2023 · 549270f · 549270f
1 parent fbf2a66
commit 549270f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/examples/uncollapsed_vi.py b/docs/examples/uncollapsed_vi.py
@@ -256,7 +256,7 @@
 # %%
 schedule = ox.warmup_cosine_decay_schedule(
     init_value=0.0,
-    peak_value=0.1,
+    peak_value=0.01,
     warmup_steps=75,
     decay_steps=1500,
     end_value=0.001,