easydel.trainers.trainer.modeling_output

easydel.trainers.trainer.modeling_output#

class easydel.trainers.trainer.modeling_output.TrainerOutput(state: Any, mesh: Optional[jax._src.mesh.Mesh], checkpoint_manager: Any, gather_fns: Union[Any, Mapping[str, Callable], Dict[str, Callable], NoneType] = None, shard_fns: Union[Any, Mapping[str, Callable], Dict[str, Callable], NoneType] = None, last_save_file_name: Optional[str] = None, checkpoint_path: Optional[str] = None)[source]#

Bases: object

checkpoint_manager: Any#
checkpoint_path: Optional[str] = None#
classmethod from_dict(data: Dict[str, Any]) → T#

Deserializes a dictionary into a PyTree object.

classmethod from_json(json_str: str) → T#

Deserializes a JSON string into a PyTree object.

gather_fns: Optional[Union[Any, Mapping[str, Callable], Dict[str, Callable]]] = None#
last_save_file_name: Optional[str] = None#
mesh: Optional[Mesh]#
replace(**kwargs)#

Creates a new instance with specified fields replaced.

shard_fns: Optional[Union[Any, Mapping[str, Callable], Dict[str, Callable]]] = None#
state: Any#
to_dict() → Dict[str, Any]#

Serializes the PyTree object to a dictionary.

to_json(**kwargs) → str#

Serializes the PyTree object to a JSON string.

Contents