So I am assuming your eps_weight is getting updated by backpropagation, right ? Can  you please share a snippet of your code to enable this.
RIght now, I was thinking of adding to x.data the eps_weight, however, that won’t let me backpropagate I guess to the eps_weight