2 Commits
Author SHA1 Message Date
Michael Tu 8a00f9cdcd Make Kim CNN ONNX-exportable (#136)
* Kim CNN - only set embedding for corresponding mode

* Kim CNN ONNX Export

* Specify dummy ONNX input size from command line
2018-08-04 17:30:24 -04:00
Michael Tu 8563ad5976 Kim CNN OOP Refactoring (#124)
* Nuke obsolete artifacts

* Refactor Kim CNN

* Make kim_cnn a module

* Fix bugs

* Update README

* Add choices to dataset arg

* update for sst2

update sst.py

update sst.py

* Add Kim CNN dataset choices to args.py

* Update tuned SST-1 accuracy
2018-07-03 16:42:31 -04:00