habitat.analysis.mlp.mlp

Module Contents

Classes

MLPBase
LinearMLP
LSTMMLP
Conv2DMLP
ConvTranspose2DMLP
BMMMLP
RuntimePredictor

class habitat.analysis.mlp.mlp.MLPBase(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.LinearMLP(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.LSTMMLP(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.Conv2DMLP(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.ConvTranspose2DMLP(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.BMMMLP(layers, layer_size)

Bases: torch.nn.Module

forward(x)

class habitat.analysis.mlp.mlp.RuntimePredictor(model_name, layers, layer_size, model_path=None)

load_state(path)

save_state(path)

_train()

_validate()

train_with_dataset(dataset_path, epochs=40, use_cuda=True)

predict(kernel_arguments, device_name)