Commit Graph
9 Commits
Author SHA1 Message Date
Michael Tu a363e3d256 Add flag to support onnx (#91) 2017-12-06 09:55:02 -05:00
rosequ 28a198f33c Update bridge (#87) (#89)
* initial commit of the updated bridge

* moved bridge file to the root

* after CR1

* after CR2
2017-12-05 12:23:08 -05:00
Matt Crane ed4dba2497 Typo fix (#54) 2017-09-22 11:30:26 +09:00
Matt Crane a601722768 Option to disable the CuDNN backend (#52)
Some of the kernels in this backend are known to be non-deterministic. It's also not clear to me whether it's enabled in a default install or not, enabled is set to True by default, but I think ultimately it's a runtime choice.
2017-09-22 11:19:30 +09:00
Michael Tu 245b032545 Set gpu seed for SM model (#46)
Small fix to set GPU seed for SM model
2017-09-12 14:28:18 -04:00
Matt Crane a3294339fb Add seed and thread arguments (#43)
So that any experiments can be controlled from the command line, I've set the defaults to what they were hardcoded (for the seed) and num_threads in kim_cnn to be consistent with the default in sm_cnn.
2017-09-01 09:30:01 -04:00
Michael Tu 43dd6fdb1f GPU Support for SM Model (#26)
Add code to using GPU for the SM Model (#25). To use the GPU if one is available, add the --cuda optional parameter when calling main.py.
2017-06-02 08:47:13 -04:00
gauravbaruah 9bd5b7bb2a Corpus idf (#23)
as part of sourcing-IDF-from-index and e2e experiments.
2017-05-07 22:28:35 -04:00
rosequ 212aa6fb93 renaming sm_model; faster bridge (#18) (#22)
+ renamed sm_model
+ faster bridge by obtaining the IDF scores of a term directly from the Java server
2017-05-03 17:24:47 -04:00