quantize models

[halo34]

Hi, I have some problems with torchinfo and quantize networks. Are there special input options?

[TylerYep]

The README has all of the input options we have.

In particular, take a look at input_data, input_size, dtype, and passing additional parameters through *args and **kwargs