precompile jax.random.split

2023-05-10 15:20:42 +08:00
parent 0fdc856f2d
commit 9dfa904ce5
4 changed files with 19 additions and 14 deletions
--- a/utils/default_config.json
+++ b/utils/default_config.json
@@ -3,9 +3,9 @@
        "num_inputs": 2,
        "num_outputs": 1,
        "problem_batch": 4,
-        "init_maximum_nodes": 20,
+        "init_maximum_nodes": 10,
        "expands_coe": 2,
-        "pre_compile_times": 0
+        "pre_compile_times": 2
    },
    "neat": {
        "population": {