Adaptive Tracking Stochastic Distribution Control for Two-step Neural Network Models