浏览代码

Move value network for SAC to device

/develop/action-slice
Ervin Teng 4 年前
当前提交
c675393c
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1
      ml-agents/mlagents/trainers/sac/optimizer_torch.py

1
ml-agents/mlagents/trainers/sac/optimizer_torch.py


def _move_to_device(self, device: torch.device) -> None:
self._log_ent_coef.to(device)
self.target_network.to(device)
self.value_network.to(device)
self.q_network.to(device)
def sac_q_loss(

正在加载...
取消
保存