-
Karmel Allison authored
* Try out affine layer instead of dense * Use reduce mean instead of avg pooling * Remove np * Use reduce mean instead of avg pooling * Fix axes * Cleanup * Fixing comment * Fixing tests
d5663b3c
* Try out affine layer instead of dense * Use reduce mean instead of avg pooling * Remove np * Use reduce mean instead of avg pooling * Fix axes * Cleanup * Fixing comment * Fixing tests