use jit().lower.compile in create functions

2023-05-08 02:35:04 +08:00
parent 497d89fc69
commit d4a75b9394
9 changed files with 120 additions and 77 deletions
--- a/examples/xor.py
+++ b/examples/xor.py
@@ -23,8 +23,8 @@ def evaluate(forward_func: Callable) -> List[float]:
    return fitnesses.tolist()  # returns a list


-@using_cprofile
-# @partial(using_cprofile, root_abs_path='/mnt/e/neat-jax/', replace_pattern="/mnt/e/neat-jax/")
+# @using_cprofile
+@partial(using_cprofile, root_abs_path='/mnt/e/neat-jax/', replace_pattern="/mnt/e/neat-jax/")
 def main():
    config = Configer.load_config()
    pipeline = Pipeline(config, seed=11323)