change model rewriter to confirm to neural_network's init_wt input format.