initialize methods

tensorneat/pipeline.py

@@ -82,12 +82,13 @@ class Pipeline:
         state = ini_state
         compiled_step = jax.jit(self.step).lower(ini_state).compile()
 
-        for _ in range(self.generation_limit):
+        for w in range(self.generation_limit):
 
             self.generation_timestamp = time.time()
 
             previous_pop = self.algorithm.ask(state.alg)
 
+                
             state, fitnesses = compiled_step(state)
 
             fitnesses = jax.device_get(fitnesses)
@@ -102,7 +103,13 @@ class Pipeline:
             if max(fitnesses) >= self.fitness_target:
                 print("Fitness limit reached!")
                 return state, self.best_genome
-
+            node= previous_pop[0][0][:,0]
+            node_count = jnp.sum(~jnp.isnan(node))
+            conn= previous_pop[1][0][:,0]
+            conn_count = jnp.sum(~jnp.isnan(conn))
+            if(w%5==0):
+                print("node_count",node_count)
+                print("conn_count",conn_count)
         print("Generation limit reached!")
         return state, self.best_genome