I think I don't think it's necessary to turn on enable_input_require_grads() in this task scenario which is not data augmentation. But I'm not sure if other optimizations depend on this condition, I hope someone can tell me the answer to this, thanks!