Fabrik
Fabrik copied to clipboard
Fix #316
*Added CapsNet in model zoo as requested in issue #316 . *Added support for CapsuleLayer under vision category. *Added support for MaskCapsule layer and Length layer under Utility category. *Added support for Squash layer under Activation category
@namish800 Can you also share a screenshot of imported model?
@namish800 can you resolve the conflicts and fix the issue with number of parameters.
@Ram81 I added the shape calculation logic for new layers. Now input and output shapes are correct but it is not counting number of parameters of only Capsule Layer now. I am currently trying to solve this then I will get to resolving the conflicts.
I still can't figure out how does calculate_parameter function in ide/views.py works. Can you help me figure out how this function is calculating number of parameters.
@namish800 can you test the export functionality of the model and mention it in readme also can you write import & export test for the same.
@Ram81 I have tested the export functionality of the model and mentioned about it in the readme file. if I am right I have to write the tests in Fabrik/tests/unit/keras_app/test_views.py ?
@namish800 yes you have to write tests there until we have a separate test suite for celery tasks.