habitat.analysis.mlp.mlp
#
Module Contents#
Classes#
- class habitat.analysis.mlp.mlp.MLPBase(layers, layer_size)[source]#
Bases:
torch.nn.Module
- class habitat.analysis.mlp.mlp.LinearMLP(layers, layer_size)[source]#
Bases:
torch.nn.Module
- class habitat.analysis.mlp.mlp.LSTMMLP(layers, layer_size)[source]#
Bases:
torch.nn.Module
- class habitat.analysis.mlp.mlp.Conv2DMLP(layers, layer_size)[source]#
Bases:
torch.nn.Module
- class habitat.analysis.mlp.mlp.ConvTranspose2DMLP(layers, layer_size)[source]#
Bases:
torch.nn.Module
- class habitat.analysis.mlp.mlp.BMMMLP(layers, layer_size)[source]#
Bases:
torch.nn.Module