complete forward ways

2023-07-08 20:52:31 +08:00
parent 6670c4aea1
commit 7265e33c43
2 changed files with 21 additions and 4 deletions
--- a/algorithms/neat/genome/forward.py
+++ b/algorithms/neat/genome/forward.py
@@ -97,7 +97,9 @@ def create_forward_function(config):

    if config['forward_way'] == 'single':
        return jit(forward)
-        # return jit(batch_forward)
+
+    if config['forward_way'] == 'batch':
+        return jit(batch_forward)

    elif config['forward_way'] == 'pop':
        return jit(pop_batch_forward)