capsule net notebook

labml_nn/capsule_networks/__init__.py

@@ -22,6 +22,11 @@ This file holds the implementations of the core modules of Capsule Networks.
 
 I used [jindongwang/Pytorch-CapsuleNet](https://github.com/jindongwang/Pytorch-CapsuleNet) to clarify some
 confusions I had with the paper.
+
+Here's a notebook for training a Capsule Networks on MNIST dataset.
+
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/lab-ml/nn/blob/master/labml_nn/capsule_networks/mnist.ipynb)
+[![View Run](https://img.shields.io/badge/labml-experiment-brightgreen)](https://web.lab-ml.com/run?uuid=e7c08e08586711ebb3e30242ac1c0002)
 """
 
 import torch.nn as nn