search

NVIDIA / OpenSeq2Seq

Toolkit for efficient experimentation with Speech Recognition, Text2Speech and NLP
https://nvidia.github.io/OpenSeq2Seq
Apache License 2.0
1.54k stars 372 forks source link

Checking the source file length during a Text2Text model initialization #465

Open d2sys opened 5 years ago

d2sys commented 5 years ago

Hi guys, I think it is a kind of enhancement. At inference time in text2text models, when the model is initialized, we check the length of the source dataset here If someone wants to load a model into the RAM/GPU before the source text is provided, the process fails in the next steps. I propose to move dataset length init into the getter

Thanks!