fix recompute state (#7044)

* fix recompute state

fix recompute state (#7044)
* fix recompute state
d9d46f8f · Wenyu · GitHub · 7f0c6b27 · d9d46f8f
隐藏空白更改
内联并排

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,