ai4co / rl4co

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

Mps device for Macs #111

Closed ngastzepeda closed 10 months ago

ngastzepeda commented 10 months ago

Description

Included code in notebooks.cvrptw.run to make runs on MacBooks posible.

Motivation and Context

I was not able before to run small experiments on my local machine.

Types of changes

What types of changes does your code introduce? Remove all that do not apply:

Checklist

Go over all the following points, and put an x in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!