xnn.common.models.hub.loader.PretrainedModel

class xnn.common.models.hub.loader.PretrainedModel(model, config, card, cutoff, path)

Bases: object

A loaded model with everything known about it.

Variables:

• model (ForceStressOutput) – The ready model (forces, optionally stress), in eval mode.

• config (Config) – The configuration it was built from.

• card (ModelCard or None) – Its card (None for a bare checkpoint file).

• cutoff (float) – Neighbor-list cutoff of the built model, in angstrom (a dispersion wrapper widens it beyond the core model’s radius).

• path (pathlib.Path) – The local directory or file it was loaded from.

Parameters:

• model (Module)

• config (Any)

• card (ModelCard | None)

• cutoff (float)

• path (Path)