marian-nmt/marian
1
1
Fork 0
mirror of https://github.com/marian-nmt/marian.git synced 2024-09-17 09:47:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

delete variables altogether

Browse Source
This commit is contained in:
Hieu Hoang 2021-06-14 19:07:17 -07:00
parent 82fa059a03
commit 7e6ec58507
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/layers/logits.cpp
View File

@ -117,12 +117,11 @@ Expr Logits::getFactoredLogits(size_t groupIndex,
out->val()->set(masks);
};
int currBeamSize = sel->shape()[0];
int batchSize = sel->shape()[2];
//int currBeamSize = sel->shape()[0];
//int batchSize = sel->shape()[2];
Expr lastIndices = shortlist->getIndicesExpr();
assert(lastIndices->shape()[0] == currBeamSize || lastIndices->shape()[0] == 1);
assert(lastIndices->shape()[1] == batchSize || lastIndices->shape()[1] == 1);
currBeamSize; currBeamSize;
//assert(lastIndices->shape()[0] == currBeamSize || lastIndices->shape()[0] == 1);
//assert(lastIndices->shape()[1] == batchSize || lastIndices->shape()[1] == 1);
factorMasks = lambda({lastIndices}, lastIndices->shape(), Type::float32, forward);