nce_layer cannot accept neg_distribution

roysh · Apr 12, 2017 · e8f65b8 · e8f65b8
1 parent caffcc8
commit e8f65b8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/trainer_config_helpers/layers.py b/python/paddle/trainer_config_helpers/layers.py
@@ -4873,7 +4873,7 @@ def nce_layer(input,
     if neg_distribution is not None:
         assert isinstance(neg_distribution, collections.Sequence)
         assert len(neg_distribution) == num_classes
-        assert sum(neg_distribution) == 1
+        assert abs(sum(neg_distribution) - 1.0) < 1e-5
     if not isinstance(act, BaseActivation):
         raise TypeError()