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

Replace CC `tanh` activation with `swish`

Author: awjuliani

python/trainer_config.yaml

@@ -114,10 +114,11 @@ VisualPyramidBrain:
 
 Ball3DBrain:
     normalize: true
-    batch_size: 1200
+    batch_size: 64
     buffer_size: 12000
     summary_freq: 1000
     time_horizon: 1000
+    lambd: 0.99
     gamma: 0.995
     beta: 0.001
 

python/unitytrainers/models.py

@@ -168,10 +168,7 @@ def create_observation_streams(self, num_streams, h_size, num_layers):
         :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):