save wandb model

evo/master.py

@@ -87,7 +87,7 @@ class CMAMaster:
                 {
                     "Train Score Mean": scores.mean(),
                     "Train Score Std": scores.std(),
-                    "Train Step Time Cost": time_cost/60.0,
+                    "Train Step Time Cost": time_cost / 60.0,
                     # "global_step": iter_count,
                 },
                 commit=False
@@ -126,7 +126,7 @@ class CMAMaster:
                     {
                         "Evaluation Score Mean": eval_scores.mean(),
                         "Evaluation Score Std": eval_scores.std(),
-                        "Evaluation Step Time Cost": time_cost/60.0,
+                        "Evaluation Step Time Cost": time_cost / 60.0,
                     },
                     commit=False
                     # step=iter_count,
@@ -207,10 +207,11 @@ class CMAMaster:
 
     def save_solution(self, iter_num, best):
         self.update_solution()
-        pickle.dump(
-            self.algorithm, file=open(self.log_dir + "/cma_algorithm.pickle", "wb")
-        )
+        #pickle.dump(
+        #    self.algorithm, file=open(self.log_dir + "/cma_algorithm.pickle", "wb")
+        #)
         self.solution.save(self.log_dir, iter_num, best)
+        wandb.save(os.path.join(self.log_dir, "best_model.npz"))
 
     def update_solution(self):
         self.solution.set_params(self.algorithm.get_current_parameters())