浏览代码

bug fix on range of observations

/develop-generalizationTraining-TrainerController
vincentpierre 7 年前
当前提交
b7f787f6
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      python/trainers/ppo_trainer.py

2
python/trainers/ppo_trainer.py


else:
feed_dict = {self.model.batch_size: len(info.states), self.model.sequence_length: 1}
if self.use_observations:
for i in range(info.observations):
for i in range(len(info.observations)):
feed_dict[self.model.observation_in[i]] = info.observations[i]
if self.use_states:
feed_dict[self.model.state_in] = info.states

正在加载...
取消
保存