浏览代码

Remove double setting

/develop/add-fire
Arthur Juliani 5 年前
当前提交
5d33aca7
共有 1 个文件被更改,包括 0 次插入2 次删除
  1. 2
      ml-agents/mlagents/trainers/policy/torch_policy.py

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


self.inference_dict: Dict[str, tf.Tensor] = {}
self.update_dict: Dict[str, tf.Tensor] = {}
# TF defaults to 32-bit, so we use the same here.
torch.set_default_tensor_type(torch.DoubleTensor)
reward_signal_configs = trainer_settings.reward_signals
reward_signal_names = [key.value for key, _ in reward_signal_configs.items()]

正在加载...
取消
保存