sugarme
|
a6d09580aa
|
corrected pointer receiver conversion in vision sub-packages and examples
|
2020-11-01 11:59:08 +11:00 |
|
sugarme
|
5414b6ed57
|
nn/rnn: corrected interface type conversion
|
2020-10-31 23:30:04 +11:00 |
|
sugarme
|
b69d46eae4
|
converted to pointer receiver at vision sub-package
|
2020-10-31 22:11:50 +11:00 |
|
sugarme
|
75a7d89b92
|
converted to pointer receiver at tensor APIs, tensor and nn sub-packages
|
2020-10-31 19:25:32 +11:00 |
|
sugarme
|
3cd0cf1d30
|
tensor: added SaveMultiNew to test fixing runtime error when saving trained model
|
2020-10-29 23:40:24 +11:00 |
|
sugarme
|
5edded6ca1
|
fix(nn/varstore): Load/PartialLoad methods added mismatched shape check
|
2020-08-06 09:18:39 +10:00 |
|
sugarme
|
f2fd373edf
|
chores
(cleanup)
|
2020-08-05 20:31:01 +10:00 |
|
sugarme
|
17b9e6c46f
|
fix(nn/init): fixed KaiminigUnitform Init missing case dims = 1
|
2020-08-05 17:10:17 +10:00 |
|
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
|
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
|
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
|
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
|
57e963cfe5
|
BREAKING CHANGE: make master work for travis and cpu branch
|
2020-07-26 10:30:20 +10:00 |
|
sugarme
|
38b8139bd5
|
WIP(example/char-rnn)
|
2020-07-25 19:02:48 +10:00 |
|
sugarme
|
0ff2f910f2
|
BREAKING CHANGE: switch to auto-generated
|
2020-07-22 15:56:30 +10:00 |
|
sugarme
|
cda08ca450
|
fix(nn/conv): fixed NewConv func
|
2020-07-12 06:58:27 +10:00 |
|
sugarme
|
8b05753eb4
|
fix(tensor/module): fixed and moved BatchAccuracyForLogits to nn/sequential; chore(example): clean-up
|
2020-07-11 12:53:23 +10:00 |
|
sugarme
|
8107d429b6
|
fix(nn/module): ForwardT; feat(example/cifar)
|
2020-07-08 15:19:40 +10:00 |
|
sugarme
|
ab0ebc21eb
|
fix(nn/conv): fixed conv reflect type
|
2020-07-07 14:26:57 +10:00 |
|
sugarme
|
f8ce5d0635
|
BREAKING CHANGE(nn/conv): removed pointer receiver at NewConv2D
|
2020-07-07 10:40:05 +10:00 |
|
sugarme
|
a98d8b92f4
|
WIP(vision/inception), feat(nn/conv): added conv ND
|
2020-07-06 22:34:21 +10:00 |
|
sugarme
|
ceeb95e2c5
|
feat(example/neural-style-transfer)
|
2020-07-03 22:05:49 +10:00 |
|
sugarme
|
41f3f8af2a
|
BREAKING CHANGE(nn/linear): removed pointer receiver
|
2020-07-03 11:20:52 +10:00 |
|
sugarme
|
4f57855c9b
|
fix(nn/varstore): fixed incorrect LoadPartial
|
2020-07-02 12:44:28 +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
|
c3ceb99d1a
|
WIP(vision/resnet)
|
2020-06-29 16:35:22 +10:00 |
|
sugarme
|
8f095ea43a
|
feat(nn/batch-norm): completed
|
2020-06-25 20:24:10 +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
|
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
|
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 |
|