I am trying to run train_experiment.py (located in baselines/RLLib_training) and I am getting the following error:
File RLLib_training/RailEnvRLLibWrapper.py", line 66, in reset
data, distance, agent_data = self.env.obs_builder.split_tree(tree=np.array(obs[i_agent]),
AttributeError: ‘TreeObsForRailEnv’ object has no attribute ‘split_tree’
I tried to fix it by importing the split_tree method from utils.observation_utils and setting the num_features_per_node=self.env.obs_builder.observation_dim, but then the following error appears:
ValueError: Cannot feed value of shape (7, 378) for Tensor ‘ppo_policy/obs:0’, which has shape ‘(?, 336)’
What changes do I need to make in order to run it successfully?