How can execution of forward function change according to inference time and train time?