Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 22 additions & 0 deletions python/paddle/fluid/tests/unittests/test_nll_loss.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,7 @@ class TestNLLLoss(unittest.TestCase):
def test_NLLLoss_1D_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
prog = fluid.Program()
startup_prog = fluid.Program()
Expand Down Expand Up @@ -108,6 +109,7 @@ def test_NLLLoss_1D_mean(self):
def test_NLLLoss_1D_sum(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
prog = fluid.Program()
startup_prog = fluid.Program()
Expand Down Expand Up @@ -142,6 +144,7 @@ def test_NLLLoss_1D_sum(self):
def test_NLLLoss_1D_with_weight_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
weight_np = np.random.random(size=(10, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -181,6 +184,7 @@ def test_NLLLoss_1D_with_weight_mean(self):
def test_NLLLoss_1D_with_weight_sum(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
weight_np = np.random.random(size=(10, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -221,6 +225,7 @@ def test_NLLLoss_1D_with_weight_sum(self):
def test_NLLLoss_1D_with_weight_mean_cpu(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
weight_np = np.random.random(size=(10, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -258,6 +263,7 @@ def test_NLLLoss_1D_with_weight_mean_cpu(self):
def test_NLLLoss_1D_with_weight_no_reduce_cpu(self):
np.random.seed(200)
input_np = np.random.random(size=(10, 10)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 10, size=(10, )).astype(np.int64)
weight_np = np.random.random(size=(10, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -296,6 +302,7 @@ def test_NLLLoss_1D_with_weight_no_reduce_cpu(self):
def test_NLLLoss_2D_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5)).astype(np.int64)
prog = fluid.Program()
startup_prog = fluid.Program()
Expand Down Expand Up @@ -332,6 +339,7 @@ def test_NLLLoss_2D_mean(self):
def test_NLLLoss_2D_sum(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5)).astype(np.int64)
prog = fluid.Program()
startup_prog = fluid.Program()
Expand Down Expand Up @@ -368,6 +376,7 @@ def test_NLLLoss_2D_sum(self):
def test_NLLLoss_2D_with_weight_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -410,6 +419,7 @@ def test_NLLLoss_2D_with_weight_mean(self):
def test_NLLLoss_2D_with_weight_mean_cpu(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -450,6 +460,7 @@ def test_NLLLoss_2D_with_weight_mean_cpu(self):
def test_NLLLoss_2D_with_weight_sum(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -492,6 +503,7 @@ def test_NLLLoss_2D_with_weight_sum(self):
def test_NLLLoss_in_dims_not_2or4_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5, 5)).astype(np.int64)
prog = fluid.Program()
startup_prog = fluid.Program()
Expand Down Expand Up @@ -533,6 +545,7 @@ def test_NLLLoss_in_dims_not_2or4_mean(self):
def test_NLLLoss_in_dims_not_2or4_with_weight_mean(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -580,6 +593,7 @@ def test_NLLLoss_in_dims_not_2or4_with_weight_mean(self):
def test_NLLLoss_in_dims_not_2or4_with_weight_sum(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -630,6 +644,7 @@ def test_NLLLoss_in_dims_not_2or4_with_weight_sum(self):
def test_NLLLoss_in_dims_not_2or4_with_weight_no_reduce(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -681,6 +696,7 @@ def test_NLLLoss_in_dims_not_2or4_with_weight_no_reduce(self):
def test_NLLLoss_in_dims_not_2or4_with_weight_no_reduce_cpu(self):
np.random.seed(200)
input_np = np.random.random(size=(5, 3, 5, 5, 5)).astype(np.float64)
np.random.seed(200)
label_np = np.random.randint(0, 3, size=(5, 5, 5, 5)).astype(np.int64)
weight_np = np.random.random(size=(3, )).astype(np.float64)
prog = fluid.Program()
Expand Down Expand Up @@ -736,11 +752,13 @@ def setUp(self):
np.random.seed(200)
input_np = np.random.uniform(0.1, 0.8,
self.input_shape).astype("float64")
np.random.seed(200)
label_np = np.random.randint(0, self.input_shape[1],
self.label_shape).astype("int64")
output_np, total_weight_np = nll_loss_1d(input_np, label_np)
self.inputs = {'X': input_np, 'Label': label_np}
if self.with_weight:
np.random.seed(200)
weight_np = np.random.uniform(0.1, 0.8,
self.input_shape[1]).astype("float64")
output_np, total_weight_np = nll_loss_1d(
Expand Down Expand Up @@ -778,12 +796,14 @@ def setUp(self):
np.random.seed(200)
input_np = np.random.uniform(0.1, 0.8,
self.input_shape).astype("float64")
np.random.seed(200)
label_np = np.random.randint(0, self.input_shape[1],
self.label_shape).astype("int64")
output_np = nll_loss_1d(input_np, label_np, reduction='none')
total_weight_np = np.array([0]).astype('float64')
self.inputs = {'X': input_np, 'Label': label_np}
if self.with_weight:
np.random.seed(200)
weight_np = np.random.uniform(0.1, 0.8,
self.input_shape[1]).astype("float64")
output_np, total_weight_np = nll_loss_1d(
Expand Down Expand Up @@ -865,12 +885,14 @@ def setUp(self):
np.random.seed(200)
input_np = np.random.uniform(0.1, 0.8,
self.input_shape).astype("float64")
np.random.seed(200)
label_np = np.random.randint(0, self.input_shape[1],
self.label_shape).astype("int64")
output_np = nll_loss_2d(input_np, label_np, reduction='none')
total_weight_np = np.array([0]).astype('float64')
self.inputs = {'X': input_np, 'Label': label_np}
if self.with_weight:
np.random.seed(200)
weight_np = np.random.uniform(0.1, 0.8,
self.input_shape[1]).astype("float64")
output_np, total_weight_np = nll_loss_2d(
Expand Down