capsule-networks
capsule-networks copied to clipboard
A Tensorflow implementation of Capsule Networks
from capsule.config import cfg from capsule.utils import load_mnist from capsule.capsNet import CapsNet import os import tensorflow as tf from tqdm import tqdm return : ModuleNotFoundError Traceback (most recent call last)...
"u_hat_stopped = tf.stop_gradient(u_hat, name='stop_gradient')" I don't think W can be updated in that case
@bourdakos1 Hi! The usage of GPU is lower than 50% when training process. when i run this: `python main.py`, the progress is working well, but the usage of GPU is...
Hi, Thank you for your job. I would use it, when a try to initialise capsNet() i found this error. Can you help me? `capsNet = CapsNet(is_training=cfg.is_training)` ``` UnrecognizedFlagErrorTraceback (most...
I didnt understand why we need to tile the input in routing function? Please provide some clarity. Ref: [routing function](https://github.com/bourdakos1/capsule-networks/blob/84eb67a5b56456fc0a24d7fed8b0a53982fbd1c2/capsLayer.py#L121)
In ANN and CNN, the weights are not dependent on batch_size. But here ([wij](https://github.com/bourdakos1/capsule-networks/blob/84eb67a5b56456fc0a24d7fed8b0a53982fbd1c2/capsLayer.py#L122) and [bij](https://github.com/bourdakos1/capsule-networks/blob/84eb67a5b56456fc0a24d7fed8b0a53982fbd1c2/capsLayer.py#L92)) it is tiled according to batch_size and that number of weights will be trained....
Hi, I am getting an error when trying to run your code: "failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED". I've used the GPU before, so I am not sure why I...
I have installed tensorflow-gpu on my desktop which has two high powered NVidia gpu cards and it unfortunately is not using the gpu. I have implicitly told it to import...