perfect! fix bug about jax auto recompile

add task xor-3d
This commit is contained in:
wls2002
2023-07-02 22:15:26 +08:00
parent e711146f41
commit c4d34e877b
11 changed files with 234 additions and 104 deletions

View File

@@ -2,7 +2,6 @@ import jax
import numpy as np
from configs import Configer
from algorithms.neat import Genome
from pipeline import Pipeline
xor_inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
@@ -22,10 +21,10 @@ def evaluate(forward_func):
def main():
config = Configer.load_config("xor.ini")
pipeline = Pipeline(config, seed=6)
pipeline = Pipeline(config)
nodes, cons = pipeline.auto_run(evaluate)
g = Genome(nodes, cons, config)
print(g)
# g = Genome(nodes, cons, config)
# print(g)
if __name__ == '__main__':