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)

Bases: Mapping

checkpoint_manager: Any

checkpoint_path: Optional[str] = None

from_tuple()

gather_fns: Optional[Union[Any, Mapping[str, Callable], Dict[str, Callable]]] = None

items() → a set-like object providing a view on D's items

keys() → a set-like object providing a view on D's keys

last_save_file_name: Optional[str] = None

mesh: Optional[Mesh]

replace(**kwargs)

shard_fns: Optional[Union[Any, Mapping[str, Callable], Dict[str, Callable]]] = None

state: Any

to_tuple()

values() → an object providing a view on D's values