Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
class QNetwork(network.Network):
"""Feed Forward network."""
def __init__(self,
input_tensor_spec,
action_spec,
preprocessing_layers=None,
preprocessing_combiner=None,
conv_layer_params=None,
fc_layer_params=(75, 40),
dropout_layer_params=None,
activation_fn=tf.keras.activations.relu,
kernel_initializer=None,
batch_squash=True,
dtype=tf.float32,
name='QNetwork'):
"""Creates an instance of `QNetwork`.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment