This should just return the weights.  

For timing, runtime will presumably be dominated by the training, so I think you can just time the whole thing.