diff --git a/ding/worker/learner/comm/base_comm_learner.py b/ding/worker/learner/comm/base_comm_learner.py index a7234e0e7a..4a9562888f 100644 --- a/ding/worker/learner/comm/base_comm_learner.py +++ b/ding/worker/learner/comm/base_comm_learner.py @@ -116,6 +116,7 @@ def _create_learner(self, task_info: dict) -> 'BaseLearner': # noqa setattr(learner, item, getattr(self, item)) # Set policy in created learner. policy_cfg = task_info['policy'] + policy_cfg = EasyDict(policy_cfg) learner.policy = create_policy(policy_cfg, enable_field=['learn']).learn_mode learner.setup_dataloader() return learner