Add flag to fine-tune just the top layer of a DNN.