CUDA: 3D CNNs failing some gradient checks

deeplearning4j / deeplearning4j

Suite of tools for deploying and training deep learning models using the JVM. Highlights include model import for keras, tensorflow, and onnx/pytorch, a modular and tiny c++ library for running math code and a java based math library on top of the core c++ library. Also includes samediff: a pytorch/tensorflow like library for running deep learn...

http://deeplearning4j.konduit.ai

Apache License 2.0

13.65k stars 3.83k forks source link

CUDA: 3D CNNs failing some gradient checks #5124

Closed AlexDBlack closed 6 years ago

AlexDBlack commented 6 years ago

Same tests are passing on CPU, failing on CUDA... numerical gradients are all 0s.

raver119 commented 6 years ago

Fixed

lock[bot] commented 6 years ago

This thread has been automatically locked since there has not been any recent activity after it was closed. Please open a new issue for related bugs.