During training, the previous intermediate solutions generated by the problem decomposer are replaced with the teacher-generated solutions using teacher forcing [1]}. However, the solutions generated by the problem solver are used at inference time.
