train output now takes in all prototxt info!