search

ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
https://rl4.co
MIT License
433 stars 80 forks source link

Can't train with the quickstart notebook #208

Closed Farouc closed 2 weeks ago

Farouc commented 2 months ago

I have tried to run the quickstart notebook in colab and i couldn't train the model.

The error is the following: bug

fedebotu commented 2 months ago

Hi @Farouc!

This could be due to the updated version of Tensordict. Could you try installing rl4co from source?

!pip install -U git+https://github.com/ai4co/rl4co.git

The reason is that Tensordict was recently updated. We will push the major release to PyPI in the next couple of days!

fedebotu commented 2 months ago

@Farouc the new version is out! Now you should be able to run the original notebook without the above installation from source :)

Farouc commented 2 months ago

Thank you for your response !!