diff --git a/examples/vectorized_core/pendulum_sac.py b/examples/vectorized_core/pendulum_ppo.py similarity index 98% rename from examples/vectorized_core/pendulum_sac.py rename to examples/vectorized_core/pendulum_ppo.py index 4622d22c..0359ac6a 100644 --- a/examples/vectorized_core/pendulum_sac.py +++ b/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):