Problems about extra_capsnets

Hello!
I downloaded the `extra_capsnets.ipynb` and try to run it.
In the Training part, no matter how many epochs it trained, the `Val accuracy` is always about 10%.
I failed to find why it doesn't work in training although I checked some instructions and the dataset.
I tried to change parameters like the `batch_size` to find the reason, and it also doesn't work.
I wonder why it happened and how I can restore it. Thank you very much!