tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix a bug in EvoX monitor

Browse Source
This commit is contained in:
WLS2002
2024-08-14 15:07:29 +08:00
committed by GitHub
parent 9cf17f8d69
commit a9c0e1bd9c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/tensorneat/common/evox_adaptors/tensorneat_monitor.py
View File

@@ -84,7 +84,7 @@ class TensorNEATMonitor(Monitor):
if self.is_save: if self.is_save:
# save best # save best
best_genome = jax.device_get((pop[0][max_idx], pop[1][max_idx])) best_genome = jax.device_get(self.best_genome)
file_name = os.path.join( file_name = os.path.join(
self.genome_dir, f"{generation}.npz" self.genome_dir, f"{generation}.npz"
) )
@@ -107,4 +107,4 @@ class TensorNEATMonitor(Monitor):
f"\tfitness: valid cnt: {len(valid_fitnesses)}, max: {max_f:.4f}, min: {min_f:.4f}, mean: {mean_f:.4f}, std: {std_f:.4f}\n", f"\tfitness: valid cnt: {len(valid_fitnesses)}, max: {max_f:.4f}, min: {min_f:.4f}, mean: {mean_f:.4f}, std: {std_f:.4f}\n",
) )
self.tensorneat_algorithm.show_details(self.alg_state, self.fitness) self.tensorneat_algorithm.show_details(self.alg_state, self.fitness)