[add] YAML for 1st stage NN