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)

Bases: gridmind.policies.parameterized.base_parameterized_policy.BaseParameterizedPolicy

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

channel_first = True

conv1

conv2

conv3

fc1

policy_logits

_get_conv_output_size(x)

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

forward(x)

add_batch_dim_if_necessary(state)

get_actions(states)

get_action(state)

get_action_prob(state, action)

update(state, action)

src.gridmind.policies.parameterized.atari.atari_policy.model