Amir Ghazizadeh

> `u = u.view(x.size(0), self.num_routes, -1)` > > the size of u is [2 8 32 3 733] but the classes is 5, I wonder how to fix the error...

> I directly run the code, and only got 10% performance on train set of MNIST dataset. For the val set, the accuracy keeps staying in 0.0000e+00. Is there anything...

Hi, is the PyTorch implementation finished? I would be grateful if you could share with me the PyTorch version.