gpu slice is very slow. fix this problem

This commit is contained in:
wls2002
2023-05-10 17:48:07 +08:00
parent 9dfa904ce5
commit 3f37d79d06
4 changed files with 84 additions and 30 deletions

View File

@@ -23,11 +23,11 @@ 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=114514)
pipeline = Pipeline(config, seed=11454)
pipeline.auto_run(evaluate)