Open songshan0321 opened 3 years ago
This get_region_box() function includes several logistic_activate operations, I wonder if there is any reason why they are not included in the model inference on GPU? Theoretically, it will be faster to offload it onto GPU instead of CPU host.
This get_region_box() function includes several logistic_activate operations, I wonder if there is any reason why they are not included in the model inference on GPU? Theoretically, it will be faster to offload it onto GPU instead of CPU host.