Skip to content
This repository was archived by the owner on Apr 2, 2020. It is now read-only.
This repository was archived by the owner on Apr 2, 2020. It is now read-only.

some questions about PooledPredictor #31

@lazylazypig

Description

@lazylazypig

Hi,
when i read the source code of PooledPredictor, in the function PooledPredictor::startPredictorThread, i noticed a commit about GPU mode:
// We tried adding weight sharing between nets on the same GPU,
// which resulted in sporadic NaN outputs of cuDNN (R5) layers.
// Removing weight sharing immediately solved this problem.
it looks like we cannot share trained layers at the GPU mode with cuDNN. Does this problem solved now, when i use cuDNN v5.1?
thanks

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