The code to train the flow model using constraint violation signal can be found in experiments/train_cv_flow.py The modified SAC implementation with $|\hat{a}|^2$ to ...