microsoft
/
CNTK
зеркало из
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

fix positive

This commit is contained in:
Rui Zhao (SPEECH) 2020-04-21 14:17:29 -07:00
Родитель a437366127
Коммит b1037df547
1 изменённых файлов: 6 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

8
Source/Math/GPUMatrixCUDAKernels.cuh
Просмотреть файл

@ -5866,8 +5866,7 @@ __global__ void _assignRNNTAlphaScore2(
tuID_2 = tuID - phoneNum; //tuID for [t-1,u] tuID_2 = tuID - phoneNum; //tuID for [t-1,u]
//alphaId_2 = alphaId - numChannels * maxPhoneNum; //alpha ID for [t-1, u] //alphaId_2 = alphaId - numChannels * maxPhoneNum; //alpha ID for [t-1, u]
probId_2 = tuID_2 * totalPhoneNum + blankTokenId; //ID for p(phi|t-1,u) probId_2 = tuID_2 * totalPhoneNum + blankTokenId; //ID for p(phi|t-1,u)
if (t == 0 && u == 0) if (t == 0 && u == 0)
{ {
alphaScore[tuID] = 0.0; alphaScore[tuID] = 0.0;
@ -6059,6 +6058,11 @@ __global__ void _AddPenaltyEos(ElemType* alphaScore,
//printf("boundary: %d\n", phoneBoundary); //printf("boundary: %d\n", phoneBoundary);
betaScore[tuID] += max((ElemType) 0.0, earlyP * (ElemType)(phoneBoundary - (int) t)) + max((ElemType) 0.0, lateP * (ElemType)((int) t - phoneBoundary - delayConstraint)); betaScore[tuID] += max((ElemType) 0.0, earlyP * (ElemType)(phoneBoundary - (int) t)) + max((ElemType) 0.0, lateP * (ElemType)((int) t - phoneBoundary - delayConstraint));
alphaScore[tuID] += max((ElemType) 0.0, earlyP * (ElemType)(phoneBoundary - (int) t)) + max((ElemType) 0.0, lateP * (ElemType)((int) t - phoneBoundary - delayConstraint)); alphaScore[tuID] += max((ElemType) 0.0, earlyP * (ElemType)(phoneBoundary - (int) t)) + max((ElemType) 0.0, lateP * (ElemType)((int) t - phoneBoundary - delayConstraint));
if (betaScore[tuID] >= 0.0f)
betaScore[tuID] = -0.0001;
if (alphaScore[tuID] >= 0.0f)
alphaScore[tuID] = -0.0001;
} }
else else
{ {