Use psuedo labels for NN training