added the weightFile for training