This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
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
904f61b7fd
0a89ad3a64
commit
3bbe7364ce
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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();
Write
Preview
Loading…
Cancel
Save