digits_continued.yaml for two-stage training