deepmd.jax.utils.serialization

Functions

deserialize_to_file(→ None)	Deserialize the dictionary to a model file.
serialize_from_file(→ dict)	Serialize the model file to a dictionary.

Module Contents

deepmd.jax.utils.serialization.deserialize_to_file(model_file: str, data: dict) → None

Deserialize the dictionary to a model file.

Parameters:

model_filestr

The model file to be saved.

datadict

The dictionary to be deserialized.

deepmd.jax.utils.serialization.serialize_from_file(model_file: str) → dict

Serialize the model file to a dictionary.

Parameters:

model_filestr

The model file to be serialized.

Returns:

dict

The serialized model data.