From 9777c83ec8e92b3f4a7e0b9da2c6e10a61b5b80b Mon Sep 17 00:00:00 2001
From: StudentAMM <32074280+StudentAMM@users.noreply.github.com>
Date: Thu, 10 Oct 2024 19:30:10 +0300
Subject: [PATCH] Update muzero.cpp

Fix resume bug in reanalyze case.
---
 muzero-cpp/muzero.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/muzero-cpp/muzero.cpp b/muzero-cpp/muzero.cpp
index 338fecc..e5fccec 100644
--- a/muzero-cpp/muzero.cpp
+++ b/muzero-cpp/muzero.cpp
@@ -211,6 +211,7 @@ bool muzero(const MuZeroConfig& config, std::function<std::unique_ptr<AbstractGa
     // Start to learn
     std::cout << "Starting learning." << std::endl;
     if (config.resume) { replay_buffer->load(); }
+    if (config.resume) { reanalyze_buffer->load(); }
     learn(config, &device_manager, replay_buffer, reanalyze_buffer, &trajectory_queue, shared_stats, &stop);
 
     // Empty the queue so that the actors can exit.
@@ -249,4 +250,4 @@ bool muzero(const MuZeroConfig& config, std::function<std::unique_ptr<AbstractGa
     return true;
 }
 
-}    // namespace muzero_cpp
\ No newline at end of file
+}    // namespace muzero_cpp