8 Commits

Author SHA1 Message Date
Daniel Povey
bb7cb82b04 Some fixes/refactoring, make parameters shared 2022-04-25 13:55:27 +08:00
Daniel Povey
0d40b4617a Add knowledge-base lookup to model 2022-04-25 13:40:47 +08:00
Daniel Povey
a359bfe504 Test with CUDA, bug fixes 2022-04-25 13:19:09 +08:00
Daniel Povey
f8c7e6ffb3 Add some training code. Seems to be training successfully... 2022-04-24 23:19:46 +08:00
Daniel Povey
df39fc6783 Fix devices 2022-04-24 22:48:52 +08:00
Daniel Povey
a266922678 First version of sampling.py, tests run. 2022-04-24 22:29:11 +08:00
Daniel Povey
fe5586e847 Change dirname 2022-04-24 19:51:27 +08:00
Daniel Povey
65cd1059f3 Init pruned2_knowledge dir 2022-04-24 19:50:22 +08:00