Skip to content

Commit 6f5d62e

Browse files
committed
fix batch size setting
1 parent c84fcba commit 6f5d62e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

TensorRT/src/TRTClientFPGA.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ void TRTClientFPGA<Client>::setup() {
3333
std::unique_ptr<nic::InferContext::Options> options;
3434
nic::InferContext::Options::Create(&options);
3535

36-
options->SetBatchSize(batchSize_);
36+
//do not set batch size (hardcoded in server)
3737
for (const auto& output : context_->Outputs()) {
3838
options->AddRawResult(output);
3939
}

0 commit comments

Comments
 (0)