Skip to content

Commit 6226b82

Browse files
authored
Merge pull request #1778 from reyoung/feature/refine_nce_layer_2
nce_layer cannot accept neg_distribution
2 parents caffcc8 + e8f65b8 commit 6226b82

File tree

1 file changed

+1
-1
lines changed
  • python/paddle/trainer_config_helpers

1 file changed

+1
-1
lines changed

python/paddle/trainer_config_helpers/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4873,7 +4873,7 @@ def nce_layer(input,
48734873
if neg_distribution is not None:
48744874
assert isinstance(neg_distribution, collections.Sequence)
48754875
assert len(neg_distribution) == num_classes
4876-
assert sum(neg_distribution) == 1
4876+
assert abs(sum(neg_distribution) - 1.0) < 1e-5
48774877
if not isinstance(act, BaseActivation):
48784878
raise TypeError()
48794879

0 commit comments

Comments
 (0)