未验证 提交 c0caf0e4 编写于 作者: T tangwei12 提交者: GitHub

fix ut for static graph (#27506)

上级 66951ab2
...@@ -15,11 +15,14 @@ ...@@ -15,11 +15,14 @@
from __future__ import print_function from __future__ import print_function
import unittest import unittest
import paddle
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.incubate.fleet.base.role_maker as role_maker import paddle.fluid.incubate.fleet.base.role_maker as role_maker
from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler import fleet
from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distributed_strategy import StrategyFactory from paddle.fluid.incubate.fleet.parameter_server.distribute_transpiler.distributed_strategy import StrategyFactory
paddle.enable_static()
# For Net # For Net
base_lr = 0.2 base_lr = 0.2
emb_lr = base_lr * 3 emb_lr = base_lr * 3
......
...@@ -20,6 +20,9 @@ import numpy as np ...@@ -20,6 +20,9 @@ import numpy as np
import paddle.fluid as fluid import paddle.fluid as fluid
import paddle.fluid.core as core import paddle.fluid.core as core
import paddle
paddle.enable_static()
class TestLookupTableFuseOp(unittest.TestCase): class TestLookupTableFuseOp(unittest.TestCase):
def test_fuse(self): def test_fuse(self):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册