Learning to swim in potential flow via reinforcement learning
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. Packages including OpenAI gym, PyTorch, NumPy, Scipy need to be installed.
Please contact the author at [email protected] for more details