Ni Hui updated MXNET-97:
    Component/s: MXNet Engine

> implement DepthwiseConv2dBackwardFilterKernel from tensorflow codebase
> ----------------------------------------------------------------------
>                 Key: MXNET-97
>                 URL: https://issues.apache.org/jira/browse/MXNET-97
>             Project: Apache MXNet
>          Issue Type: Improvement
>          Components: MXNet Engine
>            Reporter: Ni Hui
>            Priority: Major
> The current mxnet implementation calls __syncthreads() function too much, 
> which is extemely slow.
> The new code comes from tensorflow, but the variable names are adjusted for 
> consistency.
> My model uses depthwise conv heavily, and now its training time per iteration 
> is over 5x faster on single P40 gpu. ( old 92s vs new 18s )

This message was sent by Atlassian JIRA

To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org
For additional commands, e-mail: issues-h...@mxnet.apache.org

Reply via email to