sugarme
|
d392e55598
|
chore(colab update
|
2020-07-05 13:41:56 +10:00 |
|
sugarme
|
124285c8f2
|
chore(colab experiment)
|
2020-07-05 13:33:42 +10:00 |
|
sugarme
|
97b8f9a653
|
chore(example/neural-style-transfer): twist to test Colab GPU
|
2020-07-05 10:02:32 +10:00 |
|
sugarme
|
1eb4f57f6f
|
chore(example/neural-style-transfer): GPU device for colab test
|
2020-07-05 01:44:34 +10:00 |
|
sugarme
|
8fcda3e1f9
|
chore(example/neural-style-transfer): updated
|
2020-07-04 12:25:53 +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
|
4b91f15865
|
feat(vision/vgg): added vgg models
|
2020-07-03 09:50:02 +10:00 |
|
sugarme
|
3c115ee79f
|
fix(tensor): free C memory at tensor.Values() method. fix(vision/resnet): free C memory
|
2020-07-02 21:30:45 +10:00 |
|
sugarme
|
9ad62a1a26
|
fix(vision/resnet): fixed mem blow out
|
2020-07-02 17:40:29 +10:00 |
|
sugarme
|
d23a606a64
|
feat(example/transfer-learning): completed
|
2020-07-02 16:26:54 +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
|
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 |
|