mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 13:12:46 +08:00
Sergey Kolesnikov
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import tensorflow as tf
|
|
import tensorflow.contrib.slim as slim
|
|
|
|
from ray.rllib.models.model import Model
|
|
from ray.rllib.models.misc import normc_initializer, get_activation_fn
|
|
|
|
|
|
class FullyConnectedNetwork(Model):
|
|
"""Generic fully connected network."""
|
|
|
|
def _build_layers(self, inputs, num_outputs, options):
|
|
hiddens = options.get("fcnet_hiddens", [256, 256])
|
|
activation = get_activation_fn(options.get("fcnet_activation", "tanh"))
|
|
|
|
with tf.name_scope("fc_net"):
|
|
i = 1
|
|
last_layer = inputs
|
|
for size in hiddens:
|
|
label = "fc{}".format(i)
|
|
last_layer = slim.fully_connected(
|
|
last_layer, size,
|
|
weights_initializer=normc_initializer(1.0),
|
|
activation_fn=activation,
|
|
scope=label)
|
|
i += 1
|
|
label = "fc_out"
|
|
output = slim.fully_connected(
|
|
last_layer, num_outputs,
|
|
weights_initializer=normc_initializer(0.01),
|
|
activation_fn=None, scope=label)
|
|
return output, last_layer
|