| {"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": 1761710117720696180, "commit_timestamp_nsecs": 1761710121755118954, "custom_metadata": {}} |