Skip to content
Snippets Groups Projects
Commit 92b4f31c authored by Nishant's avatar Nishant
Browse files

Fixed inferBatch ValueError.

parent e8f0b87c
No related branches found
No related tags found
No related merge requests found
...@@ -260,14 +260,14 @@ class Model: ...@@ -260,14 +260,14 @@ class Model:
ctcInput = evalRes[1] ctcInput = evalRes[1]
evalList = self.lossPerElement evalList = self.lossPerElement
feedDict = {self.savedCtcInput : ctcInput, self.gtTexts : sparse, self.seqLen : [Model.maxTextLen] * numBatchElements, self.is_train: False} feedDict = {self.savedCtcInput : ctcInput, self.gtTexts : sparse, self.seqLen : [Model.maxTextLen] * numBatchElements, self.is_train: False}
#lossVals = self.sess.run(evalList, feedDict) lossVals = self.sess.run(evalList, feedDict)
#probs = np.exp(-lossVals) probs = np.exp(-lossVals)
# dump the output of the NN to CSV file(s) # dump the output of the NN to CSV file(s)
if self.dump: if self.dump:
self.dumpNNOutput(evalRes[1]) self.dumpNNOutput(evalRes[1])
return (texts) return (texts, probs)
def save(self): def save(self):
......
...@@ -143,4 +143,3 @@ def main(): ...@@ -143,4 +143,3 @@ def main():
if __name__ == '__main__': if __name__ == '__main__':
main() main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment