Make torch as necessary part only for cv or nlp tasks

sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)

https://developers.sber.ru/portal/products/lightautoml

Apache License 2.0

1.14k stars 52 forks source link

Make torch as necessary part only for cv or nlp tasks #3

Closed AzurTheOwl closed 2 years ago

AzurTheOwl commented 2 years ago

What's now: Any LAMA installation also installing torch as dependency, and this takes more than 800MB additional space.

Trouble example: I use LAMA for my project with tabular data and whenever project image is created - it takes up twice more space (1.6 GB) for features I don't ever need in this project.

Idea: Make additional extras for installation ([no-torch] or something like that) or remove torch from [report] installation.

alexmryzhkov commented 2 years ago

Hi @AzurTheOwl,

Torch is the main dependency for LightAutoML because of linear model based on Torch.

To decrease the size of installation you can manually install CPU version of Torch library beforehand (but in this case you will be unable to solve NLP and CV tasks using neural networks).

Alex