12 Commits

Author SHA1 Message Date
Daniel Povey
3ba081e6d9 Add more custom_fwd,custom_bwd' 2022-04-25 23:58:34 +08:00
Daniel Povey
2c4478b6d1 Fix for half precision 2022-04-25 23:03:34 +08:00
Daniel Povey
e718c7ac88 Remove unnecessary copy 2022-04-25 20:41:00 +08:00
Daniel Povey
f6619a0b20 Remove unnecessary check 2022-04-25 20:37:06 +08:00
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