Fix requires_grad of input must be true for activation checkpoint layer in pipeline train.

Fix requires_grad of input must be true for activation checkpoint layer in pipeline train. #6055