You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository was archived by the owner on Apr 2, 2020. It is now read-only.
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
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