Skip to content

cuDNN max-pooling is not compatible with in-place dropout #2688

@mfigurnov

Description

@mfigurnov

This is the same issue as #117, except that now the problem arises only when max-pooling layer uses cuDNN engine. The reason is probably the same: cuDNN's max-pooling compares inputs and outputs of the layer to perform backward pass.

How to reproduce: train @mavenlin's network for CIFAR-10 using cuDNN-enabled Caffe. The training gets stuck after few thousand iterations. However, if pool1 layer is switched to Caffe engine, the training proceeds just fine.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions