fix: fix all zero initialized tensor problem for vit (#794)

Co-authored-by: ChongWei905 <weichong4@huawei.com>

Author: ChongWei905

mindcv/models/vit.py

@@ -329,7 +329,9 @@ def get_num_layers(self):
     def _init_weights(self):
         w = self.patch_embed.proj.weight
         w_shape_flatted = (w.shape[0], functools.reduce(lambda x, y: x*y, w.shape[1:]))
-        w.set_data(initializer(XavierUniform(), w_shape_flatted, w.dtype).reshape(w.shape))
+        w_value = initializer(XavierUniform(), w_shape_flatted, w.dtype)
+        w_value.init_data()
+        w.set_data(w_value.reshape(w.shape))
         for _, cell in self.cells_and_names():
             if isinstance(cell, nn.Dense):
                 cell.weight.set_data(