Prepare for training on fprop.