Add code to compute the size of a model for NN (commented out).