Add some utilities for model training scripts