search

artyom-beilis / pytorch_dlprim

DLPrimitives/OpenCL out of tree backend for pytorch
http://blog.dlprimitives.org/
MIT License
277 stars 17 forks source link

efficientnet_bN and densenet161 backpropogation #4

Closed artyom-beilis closed 2 years ago

artyom-beilis commented 2 years ago

Calculation of backprop fails for these nets:

Testing  densenet161
    FAIL     od=0.00001 md=5.86444
Testing  efficientnet_b0
    FAIL     od=0.00001 md=1.40656
Testing  efficientnet_b4
    FAIL     od=0.00001 md=0.20620
artyom-beilis commented 2 years ago

Fixed in https://github.com/artyom-beilis/dlprimitives/commit/8b159bb4737827c15ecc661b2a3918ddded167f1 and in https://github.com/artyom-beilis/pytorch_dlprim/commit/8a4dc83918a6b9e81c02d3a95c0527d7ac90688e