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