capsule-networks icon indicating copy to clipboard operation
capsule-networks copied to clipboard

Why the dimensions of weights are dependent on batch_size?

Open meet-minimalist opened this issue 6 years ago • 0 comments

In ANN and CNN, the weights are not dependent on batch_size. But here (wij and bij) it is tiled according to batch_size and that number of weights will be trained.

What if we want to input a single image and get its classification. The whole model needs to be reconstructed and retrained for that batch_size=1?

meet-minimalist avatar Oct 25 '18 13:10 meet-minimalist