diff --git a/RLGymPPO_CPP/src/public/RLGymPPO_CPP/Learner.cpp b/RLGymPPO_CPP/src/public/RLGymPPO_CPP/Learner.cpp index 923de67..8b23553 100644 --- a/RLGymPPO_CPP/src/public/RLGymPPO_CPP/Learner.cpp +++ b/RLGymPPO_CPP/src/public/RLGymPPO_CPP/Learner.cpp @@ -604,6 +604,7 @@ void RLGPC::Learner::Learn() { RG_LOG("Learner: Timestep limit of " << config.timestepLimit << " reached, stopping"); RG_LOG("\tStopping agents..."); agentMgr->StopAgents(); + Save(); // Save checkpoint after target timestep limit } void RLGPC::Learner::AddNewExperience(GameTrajectory& gameTraj, Report& report) { @@ -742,4 +743,4 @@ RLGPC::Learner::~Learner() { delete metricSender; delete renderSender; pybind11::finalize_interpreter(); -} \ No newline at end of file +}