espaloma.nn.baselines

Classes

FreeParameterBaseline(g_ref)

Parametrize a graph by populating the parameters with free torch.nn.Parameter.

FreeParameterBaselineInitMean(g_ref)

Parametrize a graph by populating the parameters with free torch.nn.Parameter.