gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add cnn_lnlstm layer.

Browse Source
This commit is contained in:
gyunt
2019-03-22 05:27:45 +09:00
parent 8d05917c87
commit c320622fe4
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
baselines/ppo2/layers.py
View File

@@ -83,6 +83,11 @@ def ppo_cnn_lstm(nlstm=128, layer_norm=False, pad='VALID', **conv_kwargs):
return network_fn return network_fn
@register("ppo_cnn_lnlstm")
def ppo_cnn_lnlstm(nlstm=128, **conv_kwargs):
return ppo_cnn_lstm(nlstm, layer_norm=True, **conv_kwargs)
def lstm(x, m, s, scope, nh, init_scale=1.0): def lstm(x, m, s, scope, nh, init_scale=1.0):
x = tf.layers.flatten(x) x = tf.layers.flatten(x)
nin = x.get_shape()[1] nin = x.get_shape()[1]