| {"item_handlers": {"model_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "train_dataloader_state": "grain._src.python.checkpoint_handlers.CheckpointHandler", "val_dataloader_state": "grain._src.python.checkpoint_handlers.CheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1761770428868379069, "commit_timestamp_nsecs": 1761770432977023240, "custom_metadata": {}} |