- 28 11月, 2017 1 次提交
-
-
由 wanghaoshuang 提交于
1. Remove checking for num_neg_samples. 2. Fix dims of Output(Cost) and Input(Bias). 3. Renamed num_sampled_classes to num_neg_samples. 4. Add TODO for add more distribution sampler. 5. Init grad_data of bias by zero. 6. Refine comments. 7. Register a kernel for type double.
-
- 16 11月, 2017 1 次提交
-
-
由 wanghaoshuang 提交于
-
- 08 11月, 2017 1 次提交
-
-
由 wanghaoshuang 提交于
1. Add nce forward and backward kernel for CPU
-