|
|
@@ -400,7 +400,8 @@ class InferImpl : public Infer {
|
|
|
output_array_.gpu_bytes(), cudaMemcpyDeviceToHost, stream_));
|
|
|
checkRuntime(cudaStreamSynchronize(stream_));
|
|
|
|
|
|
- vector<Attribute> arrout(num_image);
|
|
|
+ vector<Attribute> arrout();
|
|
|
+ arrout.reserve(num_image);
|
|
|
|
|
|
for (int ib = 0; ib < num_image; ++ib) {
|
|
|
float *output_array_cpu = output_array_.cpu() + ib * num_classes_;
|
