pytorch-cpp icon indicating copy to clipboard operation
pytorch-cpp copied to clipboard

Published 20 hours ago • verified publisher icon warmspringwinds

preprocess_batch might cause transfer to CPU, possible bug

Open warmspringwinds opened this issue 7 years ago • 0 comments

Inspect the preprocess_batch -- it might cause the transfer to CPU in case we are working with GPU tensors.

Having fixed that will improve the performance.

https://github.com/warmspringwinds/pytorch-cpp/blob/master/src/pytorch.cpp#L1280

warmspringwinds avatar Feb 15 '18 04:02 warmspringwinds