Skip to content

Commit b97f020

Browse files
committed
fix unittest error.
1 parent 7ff689f commit b97f020

File tree

4 files changed

+8
-17
lines changed

4 files changed

+8
-17
lines changed

paddle/gserver/layers/SequenceSliceLayer.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,9 +70,8 @@ void SequenceSliceLayer::checkInputs() {
7070
const Argument& inputSeq = getInput(0);
7171
CHECK(inputSeq.hasSeq()) << "The first input of sequence slic layer "
7272
<< "must be a sequence.";
73-
// Check inputs
7473
const MatrixPtr indices1 = getInputValue(1);
75-
CHECK_EQ(indices1->getHeight(),
74+
CHECK_EQ(static_cast<size_t>(indices1->getHeight()),
7675
inputSeq.hasSubseq() ? inputSeq.getNumSubSequences()
7776
: inputSeq.getNumSequences())
7877
<< "Height of the second input should be equal to number of sequence "

python/paddle/trainer_config_helpers/layers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6242,6 +6242,7 @@ def seq_slice_layer(input, starts, ends, name=None):
62426242
name, LayerType.SEQ_SLICE, parents=[input], size=input.size)
62436243

62446244

6245+
@wrap_name_default()
62456246
@layer_support()
62466247
def kmax_sequence_score_layer(input, name=None, beam_size=1):
62476248
"""

python/paddle/trainer_config_helpers/tests/configs/protostr/test_kmax_seq_socre_layer.protostr

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,6 @@
11
type: "nn"
22
layers {
3-
name: "input"
4-
type: "data"
5-
size: 300
6-
active_type: ""
7-
}
8-
layers {
9-
name: "data"
3+
name: "input_seq"
104
type: "data"
115
size: 128
126
active_type: ""
@@ -17,7 +11,7 @@ layers {
1711
size: 1
1812
active_type: "exponential"
1913
inputs {
20-
input_layer_name: "data"
14+
input_layer_name: "input_seq"
2115
input_parameter_name: "___fc_layer_0__.w0"
2216
}
2317
bias_parameter_name: "___fc_layer_0__.wbias"
@@ -51,15 +45,14 @@ parameters {
5145
initial_strategy: 0
5246
initial_smart: false
5347
}
54-
input_layer_names: "data"
48+
input_layer_names: "input_seq"
5549
output_layer_names: "__kmax_sequence_score_layer_0__"
5650
sub_models {
5751
name: "root"
58-
layer_names: "input"
59-
layer_names: "data"
52+
layer_names: "input_seq"
6053
layer_names: "__fc_layer_0__"
6154
layer_names: "__kmax_sequence_score_layer_0__"
62-
input_layer_names: "data"
55+
input_layer_names: "input_seq"
6356
output_layer_names: "__kmax_sequence_score_layer_0__"
6457
is_recurrent_layer_group: false
6558
}

python/paddle/trainer_config_helpers/tests/configs/test_kmax_seq_socre_layer.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,7 @@
22
#coding=utf-8
33
from paddle.trainer_config_helpers import *
44

5-
data = data_layer(name='input', size=300)
6-
7-
data = data_layer(name="data", size=128)
5+
data = data_layer(name="input_seq", size=128)
86
scores = fc_layer(input=data, size=1, act=ExpActivation())
97
kmax_seq_id = kmax_sequence_score_layer(input=scores, beam_size=5)
108

0 commit comments

Comments
 (0)