|
|
|
@ -16,7 +16,8 @@ from __future__ import print_function
|
|
|
|
|
|
|
|
|
|
import unittest
|
|
|
|
|
import paddle.fluid as fluid
|
|
|
|
|
from paddle.fluid import Embedding
|
|
|
|
|
import paddle.fluid.core as core
|
|
|
|
|
from paddle.fluid.dygraph.nn import Embedding
|
|
|
|
|
import paddle.fluid.framework as framework
|
|
|
|
|
from paddle.fluid.optimizer import SGDOptimizer
|
|
|
|
|
from paddle.fluid.dygraph.base import to_variable
|
|
|
|
@ -43,7 +44,7 @@ class SimpleLSTMRNN(fluid.Layer):
|
|
|
|
|
self.cell_array = []
|
|
|
|
|
self.hidden_array = []
|
|
|
|
|
|
|
|
|
|
def _build_once(self, input_embedding, init_hidden=None, init_cell=None):
|
|
|
|
|
def build_once(self, input_embedding, init_hidden=None, init_cell=None):
|
|
|
|
|
self.weight_1_arr = []
|
|
|
|
|
self.weight_2_arr = []
|
|
|
|
|
self.bias_arr = []
|
|
|
|
@ -175,7 +176,7 @@ class PtbModel(fluid.Layer):
|
|
|
|
|
default_initializer=fluid.initializer.UniformInitializer(
|
|
|
|
|
low=-self.init_scale, high=self.init_scale))
|
|
|
|
|
|
|
|
|
|
def _build_once(self, input, label, init_hidden, init_cell):
|
|
|
|
|
def build_once(self, input, label, init_hidden, init_cell):
|
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
def forward(self, input, label, init_hidden, init_cell):
|
|
|
|
@ -277,7 +278,8 @@ class TestDygraphPtbRnn(unittest.TestCase):
|
|
|
|
|
num_steps=num_steps,
|
|
|
|
|
init_scale=init_scale)
|
|
|
|
|
|
|
|
|
|
exe = fluid.Executor(fluid.CPUPlace())
|
|
|
|
|
exe = fluid.Executor(fluid.CPUPlace(
|
|
|
|
|
) if not core.is_compiled_with_cuda() else fluid.CUDAPlace(0))
|
|
|
|
|
sgd = SGDOptimizer(learning_rate=1e-3)
|
|
|
|
|
x = fluid.layers.data(
|
|
|
|
|
name="x", shape=[-1, num_steps, 1], dtype='int64')
|
|
|
|
@ -331,18 +333,15 @@ class TestDygraphPtbRnn(unittest.TestCase):
|
|
|
|
|
static_param_updated[static_param_name_list[k -
|
|
|
|
|
3]] = out[k]
|
|
|
|
|
|
|
|
|
|
self.assertTrue(np.allclose(static_loss_value, dy_loss.numpy()))
|
|
|
|
|
self.assertTrue(np.allclose(static_last_cell_value, last_cell.numpy()))
|
|
|
|
|
self.assertTrue(np.array_equal(static_loss_value, dy_loss.numpy()))
|
|
|
|
|
self.assertTrue(
|
|
|
|
|
np.allclose(static_last_hidden_value, last_hidden.numpy()))
|
|
|
|
|
np.array_equal(static_last_cell_value, last_cell.numpy()))
|
|
|
|
|
self.assertTrue(
|
|
|
|
|
np.array_equal(static_last_hidden_value, last_hidden.numpy()))
|
|
|
|
|
for key, value in six.iteritems(static_param_init):
|
|
|
|
|
# print("static_init name: {}, value {}".format(key, value))
|
|
|
|
|
# print("dy_init name: {}, value {}".format(key, dy_param_init[key]))
|
|
|
|
|
self.assertTrue(np.allclose(value, dy_param_init[key]))
|
|
|
|
|
self.assertTrue(np.array_equal(value, dy_param_init[key]))
|
|
|
|
|
for key, value in six.iteritems(static_param_updated):
|
|
|
|
|
# print("static name: {}, value {}".format(key, value))
|
|
|
|
|
# print("dy name: {}, value {}".format(key, dy_param_updated[key]))
|
|
|
|
|
self.assertTrue(np.allclose(value, dy_param_updated[key]))
|
|
|
|
|
self.assertTrue(np.array_equal(value, dy_param_updated[key]))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|