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.