diff --git a/mindspore_rl/algorithm/ppo/ppo_trainer.py b/mindspore_rl/algorithm/ppo/ppo_trainer.py index e258ac0..0c4b560 100644 --- a/mindspore_rl/algorithm/ppo/ppo_trainer.py +++ b/mindspore_rl/algorithm/ppo/ppo_trainer.py @@ -14,12 +14,13 @@ # ============================================================================ """PPO Trainer""" import mindspore -from mindspore import Tensor +from mindspore import Tensor, set_seed from mindspore.ops import operations as P from mindspore_rl.agent import trainer from mindspore_rl.agent.trainer import Trainer +set_seed(5) # pylint: disable=W0212 class PPOTrainer(Trainer):