If I use another dataset with different dimensions about capslayer HOT 2 CLOSED

Sorry, I didn't properly read that you already have set up data loading

I did have to make some tweaks in my fork in order to load RGB images, and also to change the number of logits on the output, specifically the num_outputs argument in the output FC layer in vectorCapsNet.py. I also wanted to experiment with setting the learning_rate for Adam Optimizer so I made it a hyper parameter, and found decent results with 0.0001 (default being learning_rate=0.001) which you can of course just manually set

I've noted the num_outputs issue here, #10

The current master of my fork isworking for the binary classification of gender on the RGB, 299x299px LFW Faces Dataset up to about 95% accuracy, but decoded reconstruction isn't great, I might need to investigate some larger Fully Connected layers in the decoding stage.

If you're still having issues could you perhaps post a picture of your TensorBoard graphs?

from capslayer.

You will need to change the parameters on line 124 of what image size to expect, but then you will need to ensure your custom data is being loaded in; have you properly set up your custom dataset loading in capslayer/utils.py?

main.py will load the dataset defined by cfg.dataset and call the appropriate dataset handler in utils.py, for example load_mnist

from capslayer.