Update Sample trainer to set output (decoding) weights.