• Deyu Fu's avatar
    Improvements to apex.mlp (#804) · 31aceeaa
    Deyu Fu authored
    * update fused bias relu backward kernel
    
    * adding support for not require first layer dgrad
    
    * fix bug: wrong layer in requires grad
    
    * add infrastructure for optional bias and activation, currently only support no bias and no relu
    
    * make bias and relu optional separately
    
    * add sigmoid activation option
    31aceeaa
mlp_cuda.cu 41.1 KB