fluid-mechanics-with-reinforcement-learning

Research during the master's degree

install

requirements:

fenicsx >= 0.5.0
stablebaselines
pytorch

start training

cd PIN_DRL
python3 train_SB3.py

由于还在读书没空管理，其实有很多没用的脚本，等闲下来之后再慢慢维护

solver

PIN-DRL/fluid_mechanics/jet_simulation.py

reinforcement learning environment

PIN-DRL/env/active_gym