deepmd.backend.pytorch

Classes

PyTorchBackend

PyTorch backend.

Module Contents

class deepmd.backend.pytorch.PyTorchBackend

Bases: deepmd.backend.backend.Backend

PyTorch backend.

name = 'PyTorch'

The formal name of the backend.

features: ClassVar[deepmd.backend.backend.Backend.Feature]

The features of the backend.

suffixes: ClassVar[list[str]] = ['.pth', '.pt']

The suffixes of the backend.

is_available() → bool

Check if the backend is available.

Returns:

bool

Whether the backend is available.

property entry_point_hook: collections.abc.Callable[[argparse.Namespace], None]

The entry point hook of the backend.

Returns:

Callable[[Namespace], None]

The entry point hook of the backend.

property deep_eval: type[deepmd.infer.deep_eval.DeepEvalBackend]

The Deep Eval backend of the backend.

Returns:

type[DeepEvalBackend]

The Deep Eval backend of the backend.

property neighbor_stat: type[deepmd.pt.utils.neighbor_stat.NeighborStat]

The neighbor statistics of the backend.

Returns:

type[NeighborStat]

The neighbor statistics of the backend.

property serialize_hook: collections.abc.Callable[[str], dict]

The serialize hook to convert the model file to a dictionary.

Returns:

Callable[[str], dict]

The serialize hook of the backend.

property deserialize_hook: collections.abc.Callable[[str, dict], None]

The deserialize hook to convert the dictionary to a model file.

Returns:

Callable[[str, dict], None]

The deserialize hook of the backend.