From da85366ac7444712177492d8307af5704a273071 Mon Sep 17 00:00:00 2001
From: fan-ziqi <fanziqi614@gmail.com>
Date: Fri, 3 Jan 2025 16:10:36 +0800
Subject: [PATCH] fix: add torch::set_num_threads(4)

---
 src/rl_sar/src/rl_sim.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/rl_sar/src/rl_sim.cpp b/src/rl_sar/src/rl_sim.cpp
index 23d5b68..0001475 100644
--- a/src/rl_sar/src/rl_sim.cpp
+++ b/src/rl_sar/src/rl_sim.cpp
@@ -32,6 +32,7 @@ RL_Sim::RL_Sim()
 
     // init rl
     torch::autograd::GradMode::set_enabled(false);
+    torch::set_num_threads(4);
     if (this->params.observations_history.size() != 0)
     {
         this->history_obs_buf = ObservationBuffer(1, this->params.num_observations, this->params.observations_history.size());