search

ZHAOZHIHAO / ClusterRouting

Pytorch implementation for the paper of "Capsule networks with non-iterative cluster routing".
MIT License
6 stars 3 forks source link

Update capsule_network.py #2

Open kdhasi opened 4 months ago

kdhasi commented 4 months ago

The number of convolutional modules was adjusted. Removes excess parameters from the network.