Commit Graph

223 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
ea87e7fa38 mnist fp16 training comparison 2023-07-07 22:50:12 +10:00
sugarme
01f1e3a7fc wired up ts.Drop() for directly free mem 2023-07-07 22:30:08 +10:00
sugarme
34e87b1302 added dtype option to nn package 2023-07-07 12:24:56 +10:00
sugarme
c095d9f7f8 removed CleanUp inside ts.NoGrad() 2023-07-06 00:20:11 +10:00
sugarme
f45f0a7ed0 changed []ts.Tensor -> []*ts.Tensor 2023-07-05 23:56:48 +10:00
sugarme
e29288f63e fetched from master 2023-07-04 23:26:20 +10:00
sugarme
4aad97b216 Merge branch 'master' of ssh://github.com/sugarme/gotch into mem 2023-07-04 23:22:04 +10:00
sugarme
afa105c5cd experiment with go GC version 1.19.4 2023-07-04 23:03:49 +10:00
Tim Cassidy
8a08a1d6ab add Progress() method to TextDataIter 2023-02-10 19:30:09 -08:00
sugarme
f8ce489c09 fixed #88 - example/char-rnn memory leak 2023-01-31 17:07:50 +11: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
36b61bf719 added more pickle types 2022-02-24 15:25:15 +11:00
sugarme
dc4ab3047c added subpackage 'pickle' and file-util 2022-02-24 12:43:39 +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
35b4732830 reworked on vision/aug 2021-06-30 21:39:40 +10:00
sugarme
fda5adc77d fixed mem leak at RandomAffine 2021-06-26 18:25:26 +10:00
sugarme
5923e0f2e2 fixed RandomAdjustSharpness 2021-06-25 13:36:25 +10:00
sugarme
8aaf69494b clean up 2021-06-25 10:06:03 +10:00
sugarme
b49afeb322 temporarily fixed aug/equalize 2021-06-24 18:41:40 +10:00
sugarme
d6bfc5cd39 WIP: aug equalize 2021-06-24 07:22:30 +10:00
sugarme
b6f5a89f7b vision/aug: added double dtype 2021-06-22 00:38:40 +10:00
sugarme
e4fd5bf140 changed vision/aug input/output tensor dtype to be uint8 2021-06-22 00:03:31 +10:00
sugarme
dbab26c5aa added libtorch APIs get/set optimizer learning rates 2021-06-06 18:04:08 +10:00
sugarme
b02185df22 added libtorch APIs GetLRs and ParamGroupNum 2021-06-04 23:07:59 +10:00
sugarme
e49f0eab20 added augment demo images 2021-05-22 21:49:31 +10:00
sugarme
7292c3575e added image augmentation and minor fixed on ts.Lstsq 2021-05-22 21:02:22 +10:00
sugarme
0c9ab73736 added example/jit-train with README.md and updated code 2021-01-02 20:34:05 +11:00
sugarme
d6fb8d88d8 example/jit-train: added 2021-01-02 16:49:36 +11:00
sugarme
2fb8acef07 added train and eval atm_ API 2021-01-02 15:40:52 +11:00
sugarme
8770fc378b Updated tensor.OfDataSize to write directly binary data; Updated ReadNpy and ReadNpz 2020-11-18 13:07:08 +11:00
sugarme
5801be4072 update to ts.SaveMultiNew 2020-11-18 01:22:36 +11:00
sugarme
b4228528bb WIP(npy): added ReadNpy and ReadNpz 2020-11-17 18:31:29 +11:00
sugarme
992ff860d4 updated example code on README and added Colab links 2020-11-15 10:32:37 +11:00
sugarme
c0da658e6b remove .ipynb_checkpoints 2020-11-15 10:03:18 +11:00
sugarme
07a2fc9979 more cleanup at example/yolo 2020-11-14 14:54:37 +11:00
sugarme
957200c17e cleanup at yolo 2020-11-14 14:51:56 +11:00
sugarme
3f2e4397c8 replace yolo notebook with a link 2020-11-14 14:43:15 +11:00
sugarme
0fe8d22c82 relocated basic notebook 2020-11-13 22:52:26 +11:00
Sugarme
faa510288f Created using Colaboratory 2020-11-13 22:44:01 +11:00
Sugarme
1c5c3699d5
Delete gotch.ipynb 2020-11-13 13:15:33 +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
8296fa1a40 Created basic Gophernotes GoTch on Google Colab 2020-11-13 11:21:25 +11:00
sugarme
c1bbee4880 tensor/print: added padding and precision 2020-11-09 17:16:34 +11:00
sugarme
b18d1cde89 tensor/print: more print options; updated for minor version 2020-11-09 15:59:25 +11:00