Renamed pendulum example properly

MushroomRL · Jan 17, 2024 · 1eaa5e3 · 1eaa5e3
1 parent 10527c8
commit 1eaa5e3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/vectorized_core/pendulum_sac.py → examples/vectorized_core/pendulum_ppo.py b/examples/vectorized_core/pendulum_sac.py → examples/vectorized_core/pendulum_ppo.py
@@ -12,6 +12,9 @@
 
 from mushroom_rl.policy import GaussianTorchPolicy
 
+torch.set_num_threads(1)
+torch.set_num_interop_threads(4)
+
 
 class Network(nn.Module):
     def __init__(self, input_shape, output_shape, n_features, **kwargs):