Sina Afrooze
93571306c3
Removed tensorflow specific code in presets ( #59 )
...
* Add generic layer specification for using in presets
* Modify presets to use the generic scheme
2018-11-06 17:39:29 +02:00
Sina Afrooze
95b4fc6888
Added ability to switch between tensorflow and mxnet using -f commandline argument. ( #48 )
...
NOTE: tensorflow framework works fine if mxnet is not installed in env, but mxnet will not work if tensorflow is not installed because of the code in network_wrapper.
2018-10-30 15:29:34 -07:00
Sina Afrooze
2046358ab0
Add docstring for architecture ( #47 )
...
- Removed get_model() from architecture because it is only implementation detail of architecture.
2018-10-30 11:02:37 +02:00
Sina Afrooze
a888226641
Move embedder, middleware, and head parameters to framework agnostic modules. ( #45 )
...
Part of #28
2018-10-29 14:46:40 -07:00
Sina Afrooze