Commit Graph

79 Commits

Author SHA1 Message Date
sugarme
4a129c0c4b travis CI 2020-10-12 21:43:10 +11:00
sugarme
41f2d149b7 fixed travisCI 2020-10-12 21:30:40 +11:00
sugarme
5a4a65e35c updated Travis CI 2020-10-12 21:21:37 +11:00
sugarme
299793804a added fixed lib cgo flags 2020-10-12 17:03:12 +11:00
sugarme
aee81d22b3 update scripts for GPU and CPU 2020-10-11 08:45:49 +11:00
sugarme
d55436e1ea update GPU script 2020-10-10 21:17:58 +11:00
sugarme
e5735c77dc WIP(patch): AtgSplit func 2020-09-24 09:52:31 +10:00
sugarme
3f95b46f6c clean-up lib.go 2020-07-30 15:43:46 +10:00
sugarme
57e963cfe5 BREAKING CHANGE: make master work for travis and cpu branch 2020-07-26 10:30:20 +10:00
sugarme
0ff2f910f2 BREAKING CHANGE: switch to auto-generated 2020-07-22 15:56:30 +10:00
sugarme
959a1c8a99 change(tensor): move ts.AtDevice from c_generated to tensor.go 2020-07-21 16:23:44 +10:00
sugarme
530acf0894 WIP(example/yolo) 2020-07-15 13:42:40 +10:00
sugarme
9ce34e697a WIP(example/yolo) 2020-07-14 13:41:18 +10:00
sugarme
f16ab429c9 WIP(example/yolo) 2020-07-13 22:00:03 +10:00
sugarme
600adf506a feat(tensor): unit test 2020-07-11 16:42:27 +10:00
sugarme
94c1af9671 fix(tensor/index): fixed incorrect counting of number of NewAxis 2020-07-10 11:10:26 +10:00
sugarme
8107d429b6 fix(nn/module): ForwardT; feat(example/cifar) 2020-07-08 15:19:40 +10:00
sugarme
0a7f3afe6e feat(vision/mobilenet) 2020-07-07 11:53:52 +10:00
sugarme
36c0bac2df feat(vision/densenet) 2020-07-06 18:32:43 +10:00
sugarme
3dde7e76ad WIP(vision/efficientnet) 2020-07-06 12:02:38 +10:00
sugarme
76f4b41ad1 fix(nn/varstore): fixed incorrect Load function and replaced '|' with '.' 2020-07-02 12:33:18 +10:00
sugarme
32d4a68e12 WIP(nn/varstore_test) 2020-07-02 09:05:08 +10:00
sugarme
fcbc4ca870 feat(example/jit) 2020-06-30 20:01:01 +10:00
sugarme
2b88cf2a87 WIP(vision/resnet): added more func 2020-06-29 17:56:11 +10:00
sugarme
006fb6bb62 feat(vision/imagenet): added 2020-06-29 15:42:36 +10:00
sugarme
13c94d5960 feat(tensor/jit): add wrapper functions 2020-06-27 20:32:09 +10:00
sugarme
c6f06d208a feat(tensor/jit): completed 2020-06-27 17:11:17 +10:00
sugarme
a8f105d0ae WIP(tensor/jit): completed IValue 2020-06-27 14:25:36 +10:00
sugarme
17c01795c4 WIP(tensor/jit): continue.. 2020-06-27 06:32:49 +10:00
sugarme
ea4d3967e5 WIP(tensor/jit) 2020-06-26 17:33:48 +10:00
sugarme
8f095ea43a feat(nn/batch-norm): completed 2020-06-25 20:24:10 +10:00
sugarme
dd267ebf1c feat(nn/layer-norm): completed 2020-06-25 18:15:19 +10:00
sugarme
3669367b71 feat(nn/sparse): added embedding layers 2020-06-25 16:30:00 +10:00
sugarme
67a01f1294 chorus(libtch/README): correct typo 2020-06-25 13:30:53 +10:00
sugarme
42c02b0f65 fix(nn/rnn): corrected LSTM and GRU binding function. feat(libtch/README): updated with function created multiple ctensor in C land memory 2020-06-25 12:04:18 +10:00
sugarme
9817f7393a feat(nn/rnn): added rnn.go feat(nn/conv-transpose): added conv-transpose.go 2020-06-24 18:14:34 +10:00
sugarme
9d31337b4f WIP(example/mnist): conv 2020-06-23 01:07:07 +10:00
sugarme
d480f969bb feat(nn/conv): added conv 2020-06-22 18:49:41 +10:00
sugarme
f36d2482a1 feat(tensor): add option to delete tensor after applying operator to free up C memory 2020-06-21 23:37:42 +10:00
sugarme
3f569fdcee feat(tensor): added more tensor API, WIP(example/nn, mnist/nn) 2020-06-19 15:26:42 +10:00
sugarme
a636372144 WIP(example/mnist): nn 2020-06-19 01:37:13 +10:00
sugarme
9be19702d1 WIP(example/mnist): nn 2020-06-18 17:14:48 +10:00
sugarme
b5f112e030 feat(nn/optimizer): completed 2020-06-18 15:31:30 +10:00
sugarme
5f7aec5061 feat(free C pointer): added 'defer C.free(unsafe.Pointer(ptr))' to all missing. Except for CMalloc func and Nll func 2020-06-17 16:45:59 +10:00
sugarme
830f9ad9df feat(nn/data): data interator Iter2 2020-06-16 13:39:02 +10:00
sugarme
cf61333fab WIP(example/mnist): auto-grad still not working 2020-06-16 01:59:41 +10:00
sugarme
9efc686748 feat(vision/image): completed 2020-06-15 10:25:10 +10:00
sugarme
c1221e959e WIP(vision/image.go) 2020-06-14 22:46:36 +10:00
sugarme
3b74f1fd16 feat(nn/init): completed 2020-06-14 12:51:38 +10:00
sugarme
5db0200e25 feat(nn/varstore): completed draft 2020-06-13 21:22:54 +10:00