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/algorithms/neat/species.py
+++ b/algorithms/neat/species.py
@@ -110,7 +110,7 @@ class SpeciesController:
                    new_members[species_id].append(i)
                    unspeciated[i] = False

-        # Second, slowly match the lonely population to new-created species.
+        # Second, slowly match the lonely population to new-created species.s
        # lonely genome is proved to be not compatible with any previous species, so they only need to be compared with
        # the new representatives.
        for i in range(pop_nodes.shape[0]):