Open warmspringwinds opened 6 years ago
Inspect the preprocess_batch -- it might cause the transfer to CPU in case we are working with GPU tensors.
preprocess_batch
Having fixed that will improve the performance.
https://github.com/warmspringwinds/pytorch-cpp/blob/master/src/pytorch.cpp#L1280
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