add trained weights