8 Commits

Author SHA1 Message Date
Daniel Povey
7d457a7781 Add some diagnostics 2022-04-25 19:34:19 +08:00
Daniel Povey
edaaec09cd Update backprop of sampling.py to be slightly more efficient. 2022-04-25 19:32:11 +08:00
Daniel Povey
aea116ea25 Change printing-prob, initial scales 2022-04-25 14:02:43 +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