Browse Source

!10068 Fix SampledSoftmaxLoss stuck issue

From: @jonwe
Reviewed-by: @tom__chen,@robingrosman
Signed-off-by:
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
3bbe7364ce
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/random/uniform_candidate_sampler_gpu_kernel.h

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/random/uniform_candidate_sampler_gpu_kernel.h View File

@@ -96,7 +96,7 @@ class UniformCandidateSamplerGpuKernel : public GpuKernel {
return false;
}
input_size_ = input_shape[0] * input_shape[1];
if (num_sampled_ * num_true_ + static_cast<int64_t>(input_size_) > range_max_ * num_true_) {
if (num_sampled_ + static_cast<int64_t>(input_size_) > range_max_) {
remove_accidental_hits_ = false;
}
InitSizeLists();


Loading…
Cancel
Save