Solve bug in trainable.supervised_trainer, problem with smooth_gradients flag