Update selfchess.py

selfchess.py

@@ -204,9 +204,12 @@ def main():
      print(f"Saving backup model to {CONFIG['backup_model_path']}")
      torch.save(model.state_dict(), CONFIG["backup_model_path"])
      with mp.Pool(processes=num_instances) as pool:
-        results = pool.starmap(game_gen, [(None,) for _ in range(num_games // 3)])
-        results += pool.starmap(game_gen, [(chess.WHITE,) for _ in range(num_games // 3)])
-        results += pool.starmap(game_gen, [(chess.BLACK,) for _ in range(num_games // 3)])
+        results_self = pool.starmap(game_gen, [(None,) for _ in range(num_games // 3)])
+        results_white = pool.starmap(game_gen, [(chess.WHITE,) for _ in range(num_games // 3)])
+        results_black = pool.starmap(game_gen, [(chess.BLACK,) for _ in range(num_games // 3)])
+        results = []
+        for s, w, b in zip(results_self, results_white, results_black):
+         results.extend([s, w, b])
         for batch in results:
             data, c, mc = batch
             print(f"Saving backup model to {CONFIG['backup_model_path']}")