amaze.extensions.sb3.networks

Half-hearted attempt a making a custom CNN.

Improvements are being worked on.

Classes

CustomCNN(observation_space[, features_dim])

Bare-bones attempt at using a custom CNN.

class amaze.extensions.sb3.networks.CustomCNN(observation_space: Box, features_dim: int = 256)[source]

Bare-bones attempt at using a custom CNN.

Parameters:
  • observation_space – (gym.Space)

  • features_dim – (int) Number of features extracted. This corresponds to the number of unit for the last layer.

forward(observations: Tensor) Tensor[source]

Performs one computational step