浏览代码

Add ReLU to the dense

/develop/tf2.0
Ervin Teng 5 年前
当前提交
d46b60b3
共有 1 个文件被更改,包括 3 次插入1 次删除
  1. 4
      ml-agents/mlagents/trainers/ppo/policy.py

4
ml-agents/mlagents/trainers/ppo/policy.py


super(VectorEncoder, self).__init__(**kwargs)
self.layers = []
for i in range(num_layers):
self.layers.append(tf.keras.layers.Dense(hidden_size))
self.layers.append(
tf.keras.layers.Dense(hidden_size, activation=tf.keras.activations.relu)
)
def call(self, inputs):
x = inputs

正在加载...
取消
保存