• Francisco Massa's avatar
    Memory efficient densenet (#1003) · de387e8c
    Francisco Massa authored
    * GPU efficient Densenets
    
    * removed `import math`
    
    * Changed 'efficient' to 'memory_efficient'
    
    * Add tests
    
    * Bugfix in test
    
    * Fix lint
    
    * Remove unecessary formatting
    de387e8c
test_models.py 4.9 KB