Skip to content

[FEATURE] Allow multiple variants of a given output #69

@PicoCentauri

Description

@PicoCentauri

With more and more finetuning und multimodal modal containing many heads it would be useful to select different energy heads when using a metatomic model in an engine. For example for the ASE calculator adding a new parameter head or similar to

def __init__(
self,
model: Union[FilePath, AtomisticModel],
*,
additional_outputs: Optional[Dict[str, ModelOutput]] = None,
extensions_directory=None,
check_consistency=False,
device=None,
non_conservative=False,
do_gradients_with_energy=True,
):

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions