From c1a66f7bdce26dbf1cae94f43b08db9310cd4b34 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Sun, 2 Aug 2020 20:33:32 +0200 Subject: [PATCH] Save es --- stable_baselines3/cmaes/cmaes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_baselines3/cmaes/cmaes.py b/stable_baselines3/cmaes/cmaes.py index f30de7c..2ddb829 100644 --- a/stable_baselines3/cmaes/cmaes.py +++ b/stable_baselines3/cmaes/cmaes.py @@ -188,7 +188,7 @@ class CMAES(BaseAlgorithm): self.policy.best_actor.load_from_vector(self.best_individual) # TODO: fix saving/loading - del self.es + # del self.es callback.on_training_end() return self