search

sberbank-ai-lab / LightAutoML

LAMA - automatic model creation framework
Apache License 2.0
887 stars 92 forks source link

AssertionError: Pipeline finished with 0 models for some reason #41

Closed darenr closed 3 years ago

darenr commented 3 years ago

I might have coded this incorrectly (very likley) but I'm getting the error:

AssertionError: Pipeline finished with 0 models for some reason``

Reproducible test case:

nlp.zip

lightautoml==0.2.14 huggingface-hub==0.0.8 transformers==4.6.1 python=3.9.5



LightAutoML is an amazing project, thank you
DESimakov commented 3 years ago

Do you have a GPU? We don't use a NN without the GPU due to low training speed on the CPU.

Your example works on kaggle notebook, but it has another python version. Link to kaggle kernel https://www.kaggle.com/simakov/notebookf7a1bdef0f

So if you have the GPU, the reason is python. We haven't tested LAMA with python 3.9+.

DESimakov commented 3 years ago

One more thing. If you want to specify bert model for the NN, you should do it directly:

nn_params={'opt_params': {'lr': 1e-5},
                   'max_length': 128, 'bs': 32, 'n_epochs': 1,
                   'lang': 'en',
                   'bert_model': 'distilbert-base-uncased'
                   },
alexmryzhkov commented 3 years ago

@darenr Hi Daren, does the message above fix your issue?

darenr commented 3 years ago

yes, thank you