提交 8037a997 authored 作者: AdeB's avatar AdeB

Fix bug in h_softmax related to n_outputs_per_class

上级 f90d326e
...@@ -2157,7 +2157,7 @@ def h_softmax(x, batch_size, n_outputs, W1, b1, W2, b2, ...@@ -2157,7 +2157,7 @@ def h_softmax(x, batch_size, n_outputs, W1, b1, W2, b2,
target_classes = target // n_outputs_per_class target_classes = target // n_outputs_per_class
# Outputs to which belong each target inside a class # Outputs to which belong each target inside a class
target_outputs_in_class = target % n_classes target_outputs_in_class = target % n_outputs_per_class
# Second softmax that computes the output probabilities # Second softmax that computes the output probabilities
activations = sparse_block_dot( activations = sparse_block_dot(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论