I plan to implement this paper in pytorch: Learning to Share 29. I need some advice on how to properly implement the proximal operator. It seems relate to optimizer in pytorch. Here is the equation for gradient update in this paper:

Screenshot from 2018-06-14 21-34-28.png729×118 14.7 KB

For more detail, you can check the paper. Currently, I have no clue how to do it as the gradient computation is wrapped in a proximal gradient operator.