Skip to content
Snippets Groups Projects
Commit e35487c9 authored by Franck Dary's avatar Franck Dary
Browse files

Corrected bug

parent 3c41224b
Branches
No related tags found
No related merge requests found
Show whitespace changes
Inline Side-by-side
Showing
with 2 additions and 1 deletion
decoder/src/Beam.cpp
+ 2
− 1
View file @ e35487c9
...@@ -49,7 +49,8 @@ void Beam::update(ReadingMachine & machine, bool debug) ...@@ -49,7 +49,8 @@ void Beam::update(ReadingMachine & machine, bool debug)
auto context = classifier.getNN()->extractContext(elements[index].config).back(); auto context = classifier.getNN()->extractContext(elements[index].config).back();
auto neuralInput = torch::from_blob(context.data(), {(long)context.size()}, torch::kLong).clone().to(NeuralNetworkImpl::device); auto neuralInput = torch::from_blob(context.data(), {(long)context.size()}, torch::kLong).clone().to(NeuralNetworkImpl::device);
auto prediction = torch::softmax(classifier.getNN()(neuralInput).squeeze(), 0);
auto prediction = torch::softmax(classifier.getNN()(neuralInput).squeeze(0), 0);
std::vector<std::pair<float, int>> scoresOfTransitions; std::vector<std::pair<float, int>> scoresOfTransitions;
for (unsigned int i = 0; i < prediction.size(0); i++) for (unsigned int i = 0; i < prediction.size(0); i++)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment