Skip to content

Add gradient-checkpointing option

Yin, Junqi requested to merge add_grad_ckpt into master

Adding gradient-checkpointing to fit large network. Options are 'collection', 'memory', 'speed'. It increases total number of operations.

Merge request reports