update train/prototxt to re unitialise weights