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