提交 cb08ad29 编写于 作者: K kechxu 提交者: Kecheng Xu

Prediction: remove a redundant load torch module

上级 2dafbe9c
......@@ -95,9 +95,6 @@ void JunctionMLPEvaluator::Evaluate(Obstacle* obstacle_ptr) {
torch_input[0][i] = static_cast<float>(feature_values[i]);
}
torch_inputs.push_back(torch_input.to(device_));
std::shared_ptr<torch::jit::script::Module> torch_module =
torch::jit::load(FLAGS_torch_vehicle_junction_mlp_file, device_);
std::vector<double> probability;
if (latest_feature_ptr->junction_feature().junction_exit_size() > 1) {
CHECK_NOTNULL(torch_model_ptr_);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册