Giuseppe Futia
6c3dba867a
[Tutorial] Update prerequisites of README ( #380 )
...
* Update prerequisites of README
* dependencies for pytorch models
* dependencies for mxnet models
* minor
2019-02-11 22:58:17 -05:00
VoVAllen
e17c41c0da
[Model][Tutorial] Fix capsule memory leak ( #185 )
...
* fix memory leak & Remove unnecessary initializer
* change confused name
* fix name
* Move func outside loop
* fix name inconsistency
2018-11-29 09:40:21 -05:00
VoVAllen
bf6d0025aa
[Bugfix] fix gpu manual choice for capsule example ( #164 )
2018-11-21 22:01:24 -05:00
VoVAllen
7cb50072a4
[Doc][Model] New Capsule Tutorial & Example ( #143 )
...
* new capsule tutorial
* capsule for new API
* fix deprecated API
* New tutorial and example
* investigate gc problem
* add viz code
* new capsule tutorial
* remove ipynb
* move u_hat
* add link
* add requirements.txt
* remove ani.save
* update ci to install requirements
* add graphviz
2018-11-12 12:43:13 -05:00
VoVAllen
d0ea98befa
[Model] Capsule ( #95 )
...
* add capsule example
* clean code
* better naming
* better naming
* Clean Codes based on pytorch MNIST example
* Clean codes
* Add README
2018-10-26 23:56:13 -04:00