From a75fa6d72dc69028dd2d27e65519101b01f1d4f2 Mon Sep 17 00:00:00 2001
From: Wenyu <wenyu.lyu@gmail.com>
Date: Fri, 30 Sep 2022 11:58:50 +0800
Subject: [PATCH] [cp] fix recompute state (#7045)

* fix recompute state
---
 ppdet/modeling/backbones/vision_transformer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppdet/modeling/backbones/vision_transformer.py b/ppdet/modeling/backbones/vision_transformer.py
index 26e6ff2a0..e566e6a3b 100644
--- 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,
-- 
GitLab