未验证 提交 d9d46f8f 编写于 作者: W Wenyu 提交者: GitHub

fix recompute state (#7044)

* fix recompute state
上级 7f0c6b27
...@@ -352,7 +352,8 @@ class VisionTransformer(nn.Layer): ...@@ -352,7 +352,8 @@ class VisionTransformer(nn.Layer):
self.final_norm = final_norm self.final_norm = final_norm
if use_checkpoint: if use_checkpoint:
print('please set: FLAGS_allocator_strategy=naive_best_fit') paddle.seed(0)
self.patch_embed = PatchEmbed( self.patch_embed = PatchEmbed(
img_size=img_size, img_size=img_size,
patch_size=patch_size, patch_size=patch_size,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册