src.gridmind.policies.parameterized.discrete_action_cnn_policy

Classes

DiscreteActionCNNPolicy

Helper class that provides a standard way to create an ABC using

Module Contents

class src.gridmind.policies.parameterized.discrete_action_cnn_policy.DiscreteActionCNNPolicy(observation_shape: tuple, num_actions: int)[source]

Bases: gridmind.policies.parameterized.base_parameterized_policy.BaseParameterizedPolicy

Helper class that provides a standard way to create an ABC using inheritance.

conv1[source]
conv2[source]
conv3[source]
fc1[source]
fc2[source]
_get_conv_output_size(x)[source]

Helper function to compute the size of the flattened output after convolutions.

forward(x)[source]
get_action(state)[source]
get_action_prob(state, action)[source]
update(state, action, value)[source]