xnn.common.finetune.heads.head_module_names

xnn.common.finetune.heads.head_module_names(model)

The attributes of model that make up one readout head.

Parameters:

model (torch.nn.Module) – A potential.

Returns:

The model’s head_modules.

Return type:

tuple of str

Raises:

TypeError – If the model declares none (a classical force field, say).