search

OperationalBina / PipeRT

MIT License
11 stars 7 forks source link

Delete torch #54

Closed ItayHoresh closed 4 years ago

ItayHoresh commented 4 years ago

We need to check if we can get rid of torch.

We are using torch.multiprocessing for running routine as a process.

AmitKarin commented 4 years ago

I saw in the code that in the yolov3 component we use torch_utils with more function from torch. So can we really get rid of it? Or maybe we can take just what we need for this as well?

ItayHoresh commented 4 years ago

I don't know for sure if we can get rid of it. The goal is to delete it for creating smaller image. If we must have in order to run some of our routines there is nothing we can do. @AmitKarin if you think it's not possible - say so and we will close the issue.

AmitKarin commented 4 years ago

@ItayHoresh I see in the torch_utils that the yolov3 component use the "select_device" func which can allow to select the CUDA as the device. I don't think we can delete it.

ItayHoresh commented 4 years ago

Torch is now optional.