support for weights while training is added