search

svineet / pid-simulation

PID controller simulation and subsequent control using Reinforcement Learning algorithms.
3 stars 0 forks source link

Add GPU training option #3

Open svineet opened 4 years ago

svineet commented 4 years ago

If this argument is true then shift all models to GPU