search

jina-ai / executors

internal-only
Apache License 2.0
31 stars 12 forks source link

feat(gpu): split TorchObjectDetectionSegmenter in cpu and gpu #199

Closed jacobowitz closed 2 years ago

jacobowitz commented 2 years ago

This is part of #169

It splits the TorchObjectDetectionSegmenter into cpu/gpu versions. Also the previous device implementation was incomplete (not actually creating the model on GPU), this is fixed here