Support training non-stack parameters (e.g. linear classifier)
Support training non-stack parameters (e.g. linear classifier)

nyu-mll/spinn,stanfordnlp/spinn,stanfordnlp/spinn,nyu-mll/spinn,stanfordnlp/spinn,nyu-mll/spinn