BaseNetwork

class BaseNetwork(obs_shape: tuple, **kwargs)

Bases: torch.nn.modules.module.Module, pandemonium.utilities.registrable.Registrable

ABC for all networks that allows for registration

feature_dim

dimensions of the input to the network

Type

tuple