浏览代码

Code cleanup

/develop/critic-op-lstm-currentmem
Ervin Teng 4 年前
当前提交
4fc0f93e
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 5
      ml-agents/mlagents/trainers/optimizer/torch_optimizer.py

5
ml-agents/mlagents/trainers/optimizer/torch_optimizer.py


)
# Convert to tensors
current_obs = ObsUtil.from_buffer(batch, n_obs)
current_obs = [ModelUtils.list_to_tensor(obs) for obs in current_obs]
current_obs = [
ModelUtils.list_to_tensor(obs) for obs in ObsUtil.from_buffer(batch, n_obs)
]
next_obs = [ModelUtils.list_to_tensor(obs) for obs in next_obs]
next_obs = [obs.unsqueeze(0) for obs in next_obs]

正在加载...
取消
保存