Created by: miraiwk
PR types
New features
PR changes
Operator
Related Issue
https://github.com/PaddlePaddle/Paddle/issues/27243
Describe
Hi there, I modified the code of Spectral Norm to support updating weight_u and weight_v for the operator.
There are two state parameters weight_u and weight_v in Spectral Norm.
I add a new argument fix_state to control whether to update weight_u and weight_v.
In order to implement it, two outputs UOut and VOut have been added. If fix_state is False, weight_u and weight_v will be assigned to UOut and VOut respectively.
However, I do not know how to write the unittest for this operator : ( Since the state of spectral norm operator will be updated.