use non_reentrant_checkpoint
fix requires_grad of input must be true for activation checkpoint layer in pipeline train.
#6287
Job | Run time |
---|---|
5m 25s | |
5m 25s |