未验证 提交 a3cc2d0b 编写于 作者: J JZ-LIANG 提交者: GitHub

bugfix remove fluid (#34680)

上级 898acb1a
......@@ -6031,7 +6031,7 @@ class RecomputeOptimizer(Optimizer):
self._main_program = loss.block.program
self.block = loss.block
if startup_program == None:
startup_program = fluid.default_startup_program()
startup_program = paddle.static.default_startup_program()
with program_guard(self._main_program, startup_program):
assert len(self.checkpoint_shape) > 0, (
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册