To Run our Experiments you can go to ./torchdun/models/ and run proto.ipynb

The file will run on a toy dataset as mentioned in the python notebook.