浏览代码

merge add fire

/develop/add-fire/ghost
Andrew Cohen 4 年前
当前提交
b822283f
共有 2 个文件被更改,包括 2 次插入1 次删除
  1. 1
      config/ppo/Tennis.yaml
  2. 2
      ml-agents/mlagents/trainers/policy/torch_policy.py

1
config/ppo/Tennis.yaml


behaviors:
Tennis:
framework: pytorch
trainer_type: ppo
hyperparameters:
batch_size: 2048

2
ml-agents/mlagents/trainers/policy/torch_policy.py


return self.get_current_step()
def load_weights(self, values: List[np.ndarray]) -> None:
pass
self.actor_critic.load_state_dict(values)
def init_load_weights(self) -> None:
pass

正在加载...
取消
保存