Added in check for weight files.  If there is a weight file, then load in the learned weights.  If there is no weight file, then the weights are optimalically initialized