Learning to swim in potential flow via reinforcement learning https://journals.aps.org/prfluids/abstract/10.1103/PhysRevFluids.6.050505
The code to train the three-link fish to swim to a direction is provided here. You can do the training by running PPO_fish_direction. Required packages:
- OpenAI gym
- PyTorch
- NumPy
- Scipy
Please contact the author at [email protected] for more details