Using hyperparameter-search in Trainer

Good morning @richardliaw and @sgugger.

First of all, many thanks for your work on trainer.hyperparameter_search() that will help a lot of people!

However, I think that your notebooks ( Hyperparameter Search with Transformers and Ray Tune & text_classification.ipynb) need updates. Indeed, I tried to run them with Transformers + Ray [tune] but they failed with the same problem mentioned by @sonam ( ModuleNotFoundError: No module named 'datasets_modules').

To show the problem for example with the code from Hyperparameter Search with Transformers and Ray Tune, I published a test notebook in Colab.

Could you tell us how to make them run?
Many thanks in advance.

2 Likes