Skip to content

Commit 5619f9d

Browse files
committed
Fixed advanced benchmark bug
1 parent 85b8a0d commit 5619f9d

1 file changed

Lines changed: 2 additions & 8 deletions

File tree

examples/benchmark/advanced-benchmark/defineAdvancedBenchmark.cpp

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -144,14 +144,8 @@ void adapt_config(anira::InferenceConfig& inference_config, int buffer_size, int
144144
inference_config.update_processing_spec();
145145
inference_config.set_preprocess_input_size(std::vector<size_t>{static_cast<size_t>(input_size - receptive_field)});
146146
} else if (model == 1) {
147-
#ifdef USE_LIBTORCH
148-
inference_config.set_tensor_input_shape({{buffer_size, 1, 150}}, anira::InferenceBackend::LIBTORCH);
149-
inference_config.set_tensor_output_shape({{buffer_size, 1}}, anira::InferenceBackend::LIBTORCH);
150-
#endif
151-
#ifdef USE_ONNXRUNTIME
152-
inference_config.set_tensor_input_shape({{buffer_size, 1, 150}}, anira::InferenceBackend::ONNX);
153-
inference_config.set_tensor_output_shape({{buffer_size, 1}}, anira::InferenceBackend::ONNX);
154-
#endif
147+
inference_config.set_tensor_input_shape({{buffer_size, 1, 150}});
148+
inference_config.set_tensor_output_shape({{buffer_size, 1}});
155149
#ifdef USE_TFLITE
156150
std::string model_data = inference_config.get_model_path(anira::InferenceBackend::TFLITE);
157151
size_t pos = model_data.find("256");

0 commit comments

Comments
 (0)