src.gridmind.value_estimators.base_nn_estimator

Classes

BaseNNEstimator

Module Contents

class src.gridmind.value_estimators.base_nn_estimator.BaseNNEstimator(observation_shape: tuple, num_hidden_layers: int = 0, num_outputs: int = 1, in_features: int = 64, out_features: int = 64, use_bias: bool = True)[source]

Bases: torch.nn.Module

num_hidden_layers = 0[source]

in_features = 64[source]

out_features = 64[source]

hidden_layers[source]

_create_hidden_layer(use_bias: bool)[source]

forward(x)[source]