Commit Graph

48 Commits

Author SHA1 Message Date
Goncalves Henriques, Andre (UG - Computer Science)
9257404edd Move the name of the module 2024-04-21 15:15:00 +01:00
sugarme
163e625426 fixed mem blow-up due to gradient accumulation when loading model pretrained 2023-08-12 15:46:51 +10:00
sugarme
c095d9f7f8 removed CleanUp inside ts.NoGrad() 2023-07-06 00:20:11 +10:00
sugarme
afa105c5cd experiment with go GC version 1.19.4 2023-07-04 23:03:49 +10:00
sugarme
66401b2730 updated KaimingUniform initialization and mnist CNN 2022-11-24 13:32:46 +11:00
sugarme
1c2e6f2599 change package 'tensor' to 'ts' 2022-03-12 18:20:20 +11:00
sugarme
5a6fac51f3 varstore reworked and update 2022-03-12 15:47:15 +11:00
sugarme
8baf32d860 fixed initializing CUDA device at example/mnist cnn model 2021-07-23 16:26:39 +10:00
sugarme
5d8751bfaf updated APIs changes 2021-07-23 00:54:41 +10:00
sugarme
992ff860d4 updated example code on README and added Colab links 2020-11-15 10:32:37 +11:00
Sugarme
6af627e7bf
Delete gotch_mnist.ipynb 2020-11-13 13:13:19 +11:00
Sugarme
6056096eb1 Created MNIST CNN colab notebook 2020-11-13 12:47:07 +11:00
sugarme
c38c909977 corrected Int64Option and Float64Option param functions 2020-11-02 22:35:25 +11:00
sugarme
b69d46eae4 converted to pointer receiver at vision sub-package 2020-10-31 22:11:50 +11: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
0ff2f910f2 BREAKING CHANGE: switch to auto-generated 2020-07-22 15:56:30 +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
290421a30f WIP(fix BatchAccuracyForLogits) 2020-07-11 07:10:50 +10:00
sugarme
f8ce5d0635 BREAKING CHANGE(nn/conv): removed pointer receiver at NewConv2D 2020-07-07 10:40:05 +10:00
sugarme
41f3f8af2a BREAKING CHANGE(nn/linear): removed pointer receiver 2020-07-03 11:20:52 +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
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
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
0988295a7a feat(example/mnist): updated README.md 2020-06-17 17:03:54 +10:00
sugarme
2136e06971 removed example/mnist binary built file. 2020-06-17 16:59:48 +10:00
sugarme
d838a29d2c feat(example/mnist): added README.md 2020-06-17 16:58:18 +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
20f338014d fix(tensor/NoGrad): added a hacky way to get loss func updated 2020-06-17 16:24:27 +10:00
sugarme
613cd93443 feat(example/mnist): get works when printing out tensor 2020-06-17 11:23:00 +10:00
sugarme
830f9ad9df feat(nn/data): data interator Iter2 2020-06-16 13:39:02 +10:00
sugarme
da950fe881 WIP(example/mnis): NoGrad working with anonymous function. Backward() still not working 2020-06-16 02:56:09 +10:00
sugarme
cf61333fab WIP(example/mnist): auto-grad still not working 2020-06-16 01:59:41 +10:00
sugarme
d3ad29cb53 feat(vision/mnist): completed. WIP(example/mnist) 2020-06-15 13:19:42 +10:00