浏览代码

Merge pull request #901 from Unity-Technologies/hotfix-swish

Replace CC `tanh` activation with `swish`
/hotfix-v0.9.2a
GitHub 6 年前
当前提交
1626587d
共有 2 个文件被更改,包括 3 次插入5 次删除
  1. 3
      python/trainer_config.yaml
  2. 5
      python/unitytrainers/models.py

3
python/trainer_config.yaml


Ball3DBrain:
normalize: true
batch_size: 1200
batch_size: 64
lambd: 0.99
gamma: 0.995
beta: 0.001

5
python/unitytrainers/models.py


:return: List of encoded streams.
"""
brain = self.brain
if brain.vector_action_space_type == "continuous":
activation_fn = tf.nn.tanh
else:
activation_fn = self.swish
activation_fn = self.swish
self.visual_in = []
for i in range(brain.number_visual_observations):

正在加载...
取消
保存