dist sr: support for turning off training, better training shutdown