My job was just to train neural network