I am trying to use your network but I do not see how can I save the json symbol file so I can use the model.
You only save the params file but not the symbol file that is needed for predictions. When I try the below code:
y = capsnet(mx.sym.var('data'))
y.save('%s/model.json' % model_dir)
I get the error:
AttributeError: 'CapsNet' object has no attribute 'save'