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
|
ec5be6716f
|
WIP(example/transfer-leanring)
|
2020-07-01 17:03:34 +10:00 |
|
sugarme
|
8dee081115
|
chore(example/jit): improved format
|
2020-07-01 12:59:11 +10:00 |
|
sugarme
|
98624fad6e
|
fix(vision/image): fixed resizeReserveAspectRatio
|
2020-07-01 12:37:06 +10:00 |
|
sugarme
|
fcbc4ca870
|
feat(example/jit)
|
2020-06-30 20:01:01 +10:00 |
|
sugarme
|
47283314de
|
feat(vision/resnet): completed
|
2020-06-30 03:36:53 +10:00 |
|
sugarme
|
2b88cf2a87
|
WIP(vision/resnet): added more func
|
2020-06-29 17:56:11 +10:00 |
|
sugarme
|
c3ceb99d1a
|
WIP(vision/resnet)
|
2020-06-29 16:35:22 +10:00 |
|
sugarme
|
006fb6bb62
|
feat(vision/imagenet): added
|
2020-06-29 15:42:36 +10:00 |
|
sugarme
|
b06c9f3427
|
feat(tensor/jit): unit test
|
2020-06-28 19:06:27 +10:00 |
|
sugarme
|
7af25ed6c2
|
fix(tensor/jit): fixed writing ctensor slice to C memory
|
2020-06-28 10:41:23 +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
|
88c6841dd9
|
WIP(tensor/jit): default log.Fatal for all switch cases
|
2020-06-27 14:35:34 +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
|
cdc9b548b7
|
WIP(nn/layer-norm): added
|
2020-06-25 17:40:01 +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
|
92f6e9da15
|
feat(nn/rnn_test): added unit tests for rnn
|
2020-06-25 13:29:02 +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
|
71fb5ae79b
|
fix(KaimingUniformInit): fixed incorrect init of KaimingUniform method
|
2020-06-24 12:47:10 +10:00 |
|
sugarme
|
31a3f0e587
|
feat(example/mnis): conv
|
2020-06-24 01:37:33 +10:00 |
|
sugarme
|
3e08ff3a41
|
feat(example/mnist): conv
|
2020-06-23 19:14:08 +10:00 |
|
sugarme
|
b792c6af3c
|
fix(nn/sequential): fixed wrong index at Forward method
|
2020-06-23 15:21:16 +10:00 |
|
sugarme
|
ed709027c0
|
fix(nn/sequential, linear): fixed memory held up due to undeleted middle tensors
|
2020-06-23 13:37:26 +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
|
5861f3c525
|
chorus(clean-up) and change lr in example/mnist nn model
|
2020-06-22 15:14:32 +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
|
4ffe5feb7a
|
chorus(example/mnist): cleanup. Still the memory blow-out issue
|
2020-06-21 10:57:29 +10:00 |
|
sugarme
|
e0d2e0ca7e
|
WIP(example/mnist): nn
|
2020-06-19 21:39:34 +10:00 |
|
sugarme
|
a0faf0799d
|
Fix(nn/varstore): temp fixed add not add new variable to TrainableVariable
WIP(example/mnist): nn working but take up memory. Need to free after each epoch
|
2020-06-19 20:22:51 +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
|
cbcc9fce65
|
WIP(nn/optimizer)
|
2020-06-18 13:50:34 +10:00 |
|
sugarme
|
3b6487cf31
|
feat(nn/sequential): completed
|
2020-06-18 12:44:51 +10:00 |
|
sugarme
|
b917740c89
|
removed unexpected dependecy packages
|
2020-06-18 12:15:06 +10:00 |
|
sugarme
|
77b703a67b
|
WIP(nn/sequential)
|
2020-06-18 12:06:02 +10:00 |
|
sugarme
|
569ba9df45
|
feat(nn/linear): completed, feat(nn/varstore): completed
|
2020-06-18 11:20:14 +10:00 |
|
sugarme
|
99a66e4e02
|
feat(tensor/module): added module interface in tensor package
|
2020-06-18 09:45:57 +10:00 |
|
sugarme
|
f66af53905
|
WIP(nn/module): DefaultModuleT
|
2020-06-17 20:29:31 +10:00 |
|
sugarme
|
0988295a7a
|
feat(example/mnist): updated README.md
|
2020-06-17 17:03:54 +10:00 |
|