From a4a538c4d8ecd5b94a8e98133ef3e912e4d00aaf Mon Sep 17 00:00:00 2001 From: vincentpierre Date: Fri, 14 Dec 2018 11:12:39 -0800 Subject: [PATCH] Fix on the bc_offline_training using deep copies --- ml-agents/mlagents/trainers/bc/offline_trainer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ml-agents/mlagents/trainers/bc/offline_trainer.py b/ml-agents/mlagents/trainers/bc/offline_trainer.py index f96da8e4bd..a88c70e6af 100644 --- a/ml-agents/mlagents/trainers/bc/offline_trainer.py +++ b/ml-agents/mlagents/trainers/bc/offline_trainer.py @@ -3,6 +3,7 @@ # Contains an implementation of Behavioral Cloning Algorithm import logging +import copy from mlagents.trainers.bc.trainer import BCTrainer from mlagents.trainers.demo_loader import demo_to_buffer @@ -41,8 +42,8 @@ def __init__(self, brain, trainer_parameters, training, load, seed, run_id): trainer_parameters['demo_path'], self.policy.sequence_length) - policy_brain = brain.__dict__ - expert_brain = brain_params.__dict__ + policy_brain = copy.deepcopy(brain.__dict__) + expert_brain = copy.deepcopy(brain_params.__dict__) policy_brain.pop('brain_name') expert_brain.pop('brain_name') if expert_brain != policy_brain: