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)

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

Performs one computational step