!3834 修复vit网络在pynative模式下性能较差的问题

Merge pull request !3834 from 冯一航/fix_vit_optimizer

official/cv/vit/src/optimizer.py

@@ -18,6 +18,7 @@ import numpy as np
 
 import mindspore as ms
 import mindspore.ops as ops
+from mindspore import ms_function
 from mindspore._checkparam import Validator as validator
 
 from mindspore.common.initializer import initializer
@@ -145,6 +146,7 @@ class AdamW(Optimizer):
         self.reciprocal_scale = Tensor(1.0 / loss_scale, ms.float32)
         self.clip = clip
 
+    @ms_function
     def construct(self, gradients):
         lr = self.get_lr()
         gradients = scale_grad(gradients, self.reciprocal_scale)