Hi,
I encounter some problems that when I using DataParallel
module with Variable.reinforce(r)
function, it always comes up with the error: reinforce() can be only called on outputs of stochastic functions
.
Maybe it is a problem that DataParallel
automatically invokes gather
which makes the creator
of the variable not the output of stochastic functions. I wonder if there is any solution to the problem?
Thank,
Yikang