10 Commits
Author SHA1 Message Date
Ralph Tang 3855254870 Tidy up some code (#174) 2019-02-06 13:36:48 -05:00
Victor Yang 5afb845a4b Add PIT2015 dataset (#141)
* add prediction/qrel files dump option

* fix comment

* upgrade to torchtext 0.3

* remove ngram

* update device

* minor update

* add pit2015

* revert update torchtext

* revert update torchtext

* revert update torchtext
2018-08-12 18:56:08 -04:00
Ralph Tang 62f8abec7f Add README (#110)
* Make *QA/MSRVID work with VDPWI

* Add README
2018-05-25 11:52:25 -04:00
Ralph Tang 494ce36575 Refactor VDPWI to use common API (#109) 2018-05-25 01:30:53 -04:00
Ralph Tang 73823fcc32 Make model parallel wrt batch size 2018-02-06 19:49:12 -05:00
Ralph Tang 4b266431ff Add hyperparameter tuning script 2018-02-05 17:57:52 -05:00
Ralph Tang 901ce4d6a4 Add tensorboard visualization 2018-02-05 15:25:59 -05:00
Ralph Tang b06547b36e Fix norm gradient explosion 2018-02-05 02:48:34 -05:00
Ralph Tang a110b03164 Add GPU loop unrolling for SimCube computation 2018-02-04 23:20:22 -05:00
Ralph Tang 5df6123aba Add training code 2018-02-04 21:39:25 -05:00