src.gridmind.policies.parameterized.atari.atari_policy

Attributes

model

Classes

AtariPolicy

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

Module Contents

class src.gridmind.policies.parameterized.atari.atari_policy.AtariPolicy(observation_shape, num_actions, channel_first: bool = True)[source]

Bases: gridmind.policies.parameterized.base_parameterized_policy.BaseParameterizedPolicy

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

channel_first = True[source]
conv1[source]
conv2[source]
conv3[source]
fc1[source]
policy_logits[source]
_get_conv_output_size(x)[source]

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

forward(x)[source]
add_batch_dim_if_necessary(state)[source]
get_actions(states)[source]
get_action(state)[source]
get_action_prob(state, action)[source]
update(state, action)[source]
src.gridmind.policies.parameterized.atari.atari_policy.model[source]