xnn.gnn.models.mace_foundation.load_foundation#

xnn.gnn.models.mace_foundation.load_foundation(source)[source]#

Load an upstream foundation checkpoint as a mace-torch module.

Parameters:: source (str or Path) – Alias, URL, or local path (see _checkpoint_path()).
Returns:: The unpickled upstream model, on CPU.
Return type:: torch.nn.Module
Raises:: ImportError – If mace-torch is not installed (its classes are needed to unpickle the checkpoint).