Skip to content

Commit

Permalink
fix: tensorrt does not support blank_label
Browse files Browse the repository at this point in the history
  • Loading branch information
sileht committed Sep 24, 2020
1 parent 20055d1 commit 7916500
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/backends/tensorrt/tensorrtlib.cc
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

//
#include "outputconnectorstrategy.h"
#include "tensorrtlib.h"
#include "utils/apitools.h"
Expand Down Expand Up @@ -251,7 +251,6 @@ namespace dd
cudaSetDevice(_gpuid);

APIData ad_output = ad.getobj("parameters").getobj("output");
int blank_label = -1;
std::string out_blob = "prob";
TInputConnectorStrategy inputc(this->_inputc);

Expand All @@ -266,7 +265,8 @@ namespace dd
if (_ctc)
{
if (ad_output.has("blank_label"))
blank_label = ad_output.get("blank_label").get<int>();
throw MLLibBadParamException(
"blank_label not yet implemented over tensorRT backend");
}
}

Expand Down

0 comments on commit 7916500

Please sign in to comment.