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

2
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"
) )