[cp] fix recompute state (#7045)

* fix recompute state

[cp] fix recompute state (#7045)
* fix recompute state
a75fa6d7 · Wenyu · GitHub · 7e3195b2 · a75fa6d7
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppdet/modeling/backbones/vision_transformer.py ppdet/modeling/backbones/vision_transformer.py +2 -1

未找到文件。
--- a/ppdet/modeling/backbones/vision_transformer.py
+++ b/ppdet/modeling/backbones/vision_transformer.py
@@ -352,7 +352,8 @@ class VisionTransformer(nn.Layer):
        self.final_norm = final_norm

        if use_checkpoint:
-            print('please set: FLAGS_allocator_strategy=naive_best_fit')
+            paddle.seed(0)
+
        self.patch_embed = PatchEmbed(
            img_size=img_size,
            patch_size=patch_size,