Skip to content

Commit b9ca1e0

Browse files
authored
Merge pull request #1156 from Zrachel/fixbug_cossim
fix default value of cos_sim scale
2 parents 7e03cba + 19cb2a7 commit b9ca1e0

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

python/paddle/trainer/config_parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2628,7 +2628,7 @@ def __init__(self,
26282628

26292629
@config_layer('cos')
26302630
class CosSimLayer(LayerBase):
2631-
def __init__(self, name, inputs, cos_scale=5, device=None):
2631+
def __init__(self, name, inputs, cos_scale=1, device=None):
26322632
super(CosSimLayer, self).__init__(
26332633
name, 'cos', 1, inputs=inputs, device=device)
26342634
config_assert(len(self.inputs) == 2, 'CosSimLayer must have 2 inputs')

python/paddle/trainer_config_helpers/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1673,7 +1673,7 @@ def trans_layer(input, name=None, layer_attr=None):
16731673

16741674
@wrap_name_default()
16751675
@layer_support()
1676-
def cos_sim(a, b, scale=5, size=1, name=None, layer_attr=None):
1676+
def cos_sim(a, b, scale=1, size=1, name=None, layer_attr=None):
16771677
"""
16781678
Cosine Similarity Layer. The cosine similarity equation is here.
16791679

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ layers {
7979
inputs {
8080
input_layer_name: "b"
8181
}
82-
cos_scale: 5
82+
cos_scale: 1
8383
}
8484
layers {
8585
name: "__cos_sim_1__"
@@ -92,7 +92,7 @@ layers {
9292
inputs {
9393
input_layer_name: "c"
9494
}
95-
cos_scale: 5
95+
cos_scale: 1
9696
}
9797
layers {
9898
name: "__sum_to_one_norm_layer_0__"

0 commit comments

Comments
 (0)