-
Ross Wightman authored
* Fix #2221, DenseNet issue with gradient checkpoints (memory_efficient=True) * Add grad/param count test for mem_efficient densenet
c558be6b
* Fix #2221, DenseNet issue with gradient checkpoints (memory_efficient=True) * Add grad/param count test for mem_efficient densenet