faenet.force_decoder

Module Contents

Classes

ForceDecoder
LambdaLayer

class faenet.force_decoder.ForceDecoder(type, input_channels, model_configs, act)

Bases: torch.nn.Module

forward(h)

reset_parameters()

class faenet.force_decoder.LambdaLayer(func)

Bases: torch.nn.Module

forward(x)