Cats vs dogs exmaple has issue

[shaikhsharukh]

---

TypeError Traceback (most recent call last)
in ()
----> 1 cn = CatsCaps().cuda()
2 optimizer = Adam(cn.parameters())
3 capsule_loss = CapsuleLoss().cuda()

in init(self)
13 )
14 self.primcaps = PrimaryCapsuleLayer()
---> 15 self.digicaps = SecondaryCapsuleLayer(n_capsules=2, n_routes=18432)
16 self.decoder = RegularizingDecoder(dims=[32,1024,4096,12288])
17

TypeError: init() got an unexpected keyword argument 'n_routes'