m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Polish code

Browse Source 
test=develop
revert-15470-feature/imperative
minqiyang 7 years ago
parent cafbd62ea0
commit cf7229d2c2
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File

2
python/paddle/fluid/tests/unittests/test_imperative_gan.py
Unescape View File

@ -135,7 +135,7 @@ class TestImperativeMnist(unittest.TestCase):
scope.find_var(param.name).get_tensor()) scope.find_var(param.name).get_tensor())
dy_params = dict() dy_params = dict()
with fluid.imperative.guard(): with fluid.imperative.guard(device=None):
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed

12
python/paddle/fluid/tests/unittests/test_imperative_resnet.py
Unescape View File

@ -147,7 +147,7 @@ class BottleneckBlock(fluid.imperative.Layer):
class ResNet(fluid.imperative.Layer): class ResNet(fluid.imperative.Layer):
def __init__(self, layers=50, class_dim=1000): def __init__(self, layers=50, class_dim=102):
super(ResNet, self).__init__() super(ResNet, self).__init__()
self.layers = layers self.layers = layers
@ -208,6 +208,7 @@ class TestImperativeResnet(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 1
with fluid.imperative.guard(): with fluid.imperative.guard():
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed
@ -227,7 +228,7 @@ class TestImperativeResnet(unittest.TestCase):
dy_param_init_value[param.name] = param._numpy() dy_param_init_value[param.name] = param._numpy()
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
if batch_id >= 1: if batch_id >= batch_num:
break break
dy_x_data = np.array( dy_x_data = np.array(
@ -313,7 +314,7 @@ class TestImperativeResnet(unittest.TestCase):
static_param_init_value[static_param_name_list[i]] = out[i] static_param_init_value[static_param_name_list[i]] = out[i]
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
if batch_id >= 1: if batch_id >= batch_num:
break break
static_x_data = np.array( static_x_data = np.array(
@ -368,6 +369,7 @@ class TestImperativeResnet(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 1
with fluid.imperative.guard(device=None): with fluid.imperative.guard(device=None):
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed
@ -387,7 +389,7 @@ class TestImperativeResnet(unittest.TestCase):
dy_param_init_value[param.name] = param._numpy() dy_param_init_value[param.name] = param._numpy()
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
if batch_id >= 1: if batch_id >= batch_num:
break break
dy_x_data = np.array( dy_x_data = np.array(
@ -473,7 +475,7 @@ class TestImperativeResnet(unittest.TestCase):
static_param_init_value[static_param_name_list[i]] = out[i] static_param_init_value[static_param_name_list[i]] = out[i]
for batch_id, data in enumerate(train_reader()): for batch_id, data in enumerate(train_reader()):
if batch_id >= 1: if batch_id >= batch_num:
break break
static_x_data = np.array( static_x_data = np.array(

Write Preview
Loading…
Cancel
Save