"""Training utilities for EmbodiedGen."""