Commit Graph

335 Commits

Author SHA1 Message Date
sugarme
79a0c2e6bf feat(nn/sparse): append ForwardT method for Embedding 2020-08-04 15:16:21 +10:00
sugarme
4721ffe670 fix(example/translation): more memory fix 2020-08-03 15:40:04 +10:00
sugarme
a0877826c9 WIP(example/translation): more update 2020-08-03 09:56:59 +10:00
sugarme
98ca761d30 feat(tensor/patched): add NLLLoss 2020-08-02 16:17:49 +10:00
sugarme
2bd23fcc89 WIP(example/translation): more update 2020-08-01 19:32:19 +10:00
sugarme
b77fa54eb0 WIP(example/translation) 2020-08-01 16:33:30 +10:00
sugarme
a71479e0f5 ignore lib.go 2020-07-30 15:44:31 +10:00
sugarme
3f95b46f6c clean-up lib.go 2020-07-30 15:43:46 +10:00
sugarme
2fdd5a3284 tmp open lib.go 2020-07-30 15:25:47 +10:00
sugarme
abd5c8fc7b example/char-rnn: added README.md 2020-07-28 20:10:31 +10:00
sugarme
a72ded34a3 fix(nn/rnn): fixed wrong type assertion 2020-07-28 16:15:57 +10:00
sugarme
74480db4d6 fix(nn/rnn): fixed memory leak and removed defaultSeq func; example/char-rnn: completed 2020-07-28 16:08:40 +10:00
sugarme
11a8dd9245 fix(nn/rnn): correct LSTM not flatten weights; added Device.IsCuda(); WIP(example/char-rnn): still memory blowup 2020-07-27 17:17:38 +10:00
sugarme
c69229c43a fix(tensor/data): correct NewTextData func) 2020-07-27 11:31:42 +10:00
sugarme
490ea05de2 feat(libtorch setup script) 2020-07-26 13:20:05 +10:00
sugarme
4366b60c64 change(.gitignore): ignore lib.go and dummy_cuda_dependency.cpp and fake_cuda_dependency.cpp so that we can merge between cpu and master 2020-07-26 12:39:04 +10:00
sugarme
7a90b9b478 fix(nn/varstore_test): fixed unused packages 2020-07-26 11:02:02 +10:00
sugarme
9713889246 fix: travis not working with VarStore unit test Save/load 2020-07-26 10:55:28 +10:00
sugarme
95cfc0f760 fix(.travis.yml): again 2020-07-26 10:45:45 +10:00
sugarme
8c05abca34 fix(.travis.yml) 2020-07-26 10:41:02 +10:00
sugarme
67dad18dd9 fix(.travis.yml) 2020-07-26 10:35:48 +10:00
sugarme
57e963cfe5 BREAKING CHANGE: make master work for travis and cpu branch 2020-07-26 10:30:20 +10:00
sugarme
fccf4fcc4b fix(.travis.yml) 2020-07-25 22:54:59 +10:00
sugarme
716fcd4f8c WIP(.travis.yml) 2020-07-25 22:50:08 +10:00
sugarme
d2cbeae95e chore(README): correct go report badge 2020-07-25 22:03:24 +10:00
sugarme
38b8139bd5 WIP(example/char-rnn) 2020-07-25 19:02:48 +10:00
sugarme
cddf4c8a77 feat(tensor/data_test): added some unit tests for TextData 2020-07-25 17:42:15 +10:00
sugarme
d1967817ae chore(clean up) 2020-07-25 15:19:43 +10:00
sugarme
af7655d3fc feat(tensor/data): TextData and TextDataIter; WIP(example/char-rnn) 2020-07-25 13:42:15 +10:00
sugarme
b3529b3b1b chore(example/mnist): updated README with colab link 2020-07-24 13:41:08 +10:00
Sugarme
1348830a4a updated colab - mnist 2020-07-24 13:36:26 +10:00
Sugarme
5169fe8085 Created using Colaboratory - MNIST 2020-07-24 13:32:10 +10:00
sugarme
a5c8b1918e chore(README): correct typo 2020-07-23 03:05:22 +10:00
sugarme
ea201395f4 chore(README): updated with detail example and some clean-up 2020-07-23 02:58:32 +10:00
sugarme
6df9e80f4e added go.sum 2020-07-22 17:30:14 +10:00
sugarme
829b95277c feat(license): added Apache License 2020-07-22 16:42:20 +10:00
sugarme
0ff2f910f2 BREAKING CHANGE: switch to auto-generated 2020-07-22 15:56:30 +10:00
sugarme
9f5eccb4e5 fix(DTypeFromData and FlattenData): fixed type check didn't reach data type is a slice 2020-07-22 15:26:18 +10:00
sugarme
959a1c8a99 change(tensor): move ts.AtDevice from c_generated to tensor.go 2020-07-21 16:23:44 +10:00
sugarme
579abdb750 confirmed no leak at tensor.Values(), tensor.Vals() 2020-07-17 12:32:46 +10:00
sugarme
5210385033 issue(tensor.Values): memory leak at tensor.Values() or tensor.CopyData() 2020-07-17 11:49:38 +10:00
sugarme
9fb9cd67c9 fix(example/yolo): fixed memory leak at darknet and report 2020-07-17 11:43:50 +10:00
sugarme
a272b8034d again 2020-07-17 11:32:33 +10:00
sugarme
fad3c3b53b reverse ts.Values 2020-07-17 11:22:04 +10:00
sugarme
fe9af7d4ff fix(tensor.Values): fixed wrong dtype 2020-07-17 10:49:51 +10:00
sugarme
033d797d30 change(tensor.Values): implement a new way 2020-07-17 10:43:09 +10:00
sugarme
1b07cf2c2e fix(tensor.Values): fixed memory leak 2020-07-17 10:21:45 +10:00
sugarme
4802e671a8 chore(example/yolo): cleanup 2020-07-16 16:30:11 +10:00
sugarme
55e3cc9ff5 chore(.gitignore) 2020-07-16 13:16:10 +10:00
sugarme
e1469f05fd change(example/yolo) 2020-07-16 12:49:46 +10:00