Automated PR - 2026-01-29

This commit is contained in:
sync-bot
2026-01-29 18:42:17 +00:00
parent 727c43e998
commit ca1623ad2a
31 changed files with 1723 additions and 663 deletions
@@ -128,6 +128,15 @@ class TrainingStrategy(ABC):
Scalar loss tensor
"""
def get_checkpoint_metadata(self) -> dict[str, Any]:
"""Get strategy-specific metadata to include in checkpoint files.
Override this method in subclasses to add custom metadata,
e.g. any parameters that a downstream inference pipeline may need.
Returns:
Dictionary of metadata key-value pairs (values must be JSON-serializable)
"""
return {}
def _get_video_positions(
self,
num_frames: int,