I'm trying to train an A2C model in stable-baselines3 and the EvalCallback appears to freeze when