浏览代码

Fix SAC test

/develop/critic-op-lstm-currentmem
Ervin Teng 4 年前
当前提交
180f7d03
共有 1 个文件被更改,包括 2 次插入0 次删除
  1. 2
      ml-agents/mlagents/trainers/tests/torch/test_sac.py

2
ml-agents/mlagents/trainers/tests/torch/test_sac.py


update_buffer[RewardSignalUtil.rewards_key("extrinsic")] = update_buffer[
BufferKey.ENVIRONMENT_REWARDS
]
# Mock out value memories
update_buffer[BufferKey.CRITIC_MEMORY] = update_buffer[BufferKey.MEMORY]
return_stats = optimizer.update(
update_buffer,
num_sequences=update_buffer.num_experiences // optimizer.policy.sequence_length,

正在加载...
取消
保存