浏览代码

Replace tanh on cc models w/ swish

/hotfix-v0.9.2a
Arthur Juliani 6 年前
当前提交
12d52cb0
共有 2 个文件被更改,包括 2 次插入5 次删除
  1. 2
      python/trainer_config.yaml
  2. 5
      python/unitytrainers/models.py

2
python/trainer_config.yaml


Ball3DBrain:
normalize: true
batch_size: 1200
batch_size: 64
buffer_size: 12000
summary_freq: 1000
time_horizon: 1000

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):

正在加载...
取消
保存