logichen / Neural-ODE-Robot-Learning

Cartpole dynamics prediction using neural ode solver with Gym space.

For Inference only, run: inference.py
For live training and inference from trained model, run: demo.py