Added option of not training projection layer.