lloca.backbone.mlp.MLP

class lloca.backbone.mlp.MLP(in_shape, out_shape, hidden_channels, hidden_layers, dropout_prob=None)

Bases: Module

A simple MLP.

Flattens all dimensions except batch and uses GELU nonlinearities.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

forward(inputs)

Forward pass of MLP.