unable to replicate results of example notebooks

[rohanshingade]

Can you please provide a dockerfile or requirement.txt to reproduce the results. I installed this library but had to make a few changes to run the example notebooks. On running the notebook, the number of parameters were the same as the original model. No pruning was done.