Commit Graph

104 Commits

Author SHA1 Message Date
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
d95eaba5b3 WIP: varstore rework 2022-03-11 13:21:39 +11:00
sugarme
73d6c0ae86 added more building block modules 2022-02-15 12:40:28 +11:00
sugarme
a4e5f38705 fixed nn/rnn NewLSTM() clashed weight names 2022-01-21 11:05:11 +11:00
sugarme
653caf4be5 added nn.Path.Paths() and nn.VarStore.Summary() methods 2021-11-13 22:23:38 +11:00
sugarme
880a1b25df fixed conv3D config 2021-08-03 17:56:25 +10:00
sugarme
620fccf452 added conv3dconfig 2021-08-03 12:36:10 +10:00
sugarme
dd03baa6b0 export nn/scheduler DefaultSchedulerOptions() 2021-07-25 18:29:11 +10:00
sugarme
bb00d244ed added nn/conv config options and bump version 2021-07-25 18:23:00 +10:00
sugarme
d490db3fdf added nn/schedulear NewLRScheduler 2021-07-25 17:37:44 +10:00
sugarme
5d8751bfaf updated APIs changes 2021-07-23 00:54:41 +10:00
sugarme
731513a986 fixed #45 #48 RandInit 2021-07-21 00:04:53 +10:00
sugarme
c89e4b3ba1 reworked on resnet, densenet, added BCELoss, CrossEntropyLoss, changed DataLoader.Reset() 2021-07-14 10:38:11 +10:00
sugarme
3798a51c31 fixed incorrect message at nn.Varstore.Load method 2021-06-19 18:14:15 +10:00
sugarme
efa71f3d82 added warning to varstore.LoadPartial 2021-06-19 14:31:28 +10:00
sugarme
22d9c5695a fixed multiple memory leakage at dataloader and vision image and efficientnet 2021-06-16 22:44:13 +10:00
sugarme
6f8ec3b69f added AdamW optimizer 2021-06-12 18:40:42 +10:00
sugarme
50dd5b181c added OneCycleLR 2021-06-12 18:21:43 +10:00
sugarme
2fd99cdffd WIP: OneCycleLR 2021-06-12 14:54:05 +10:00
sugarme
5f57a216c7 added CosineAnnealingWarmRestarts 2021-06-12 10:54:59 +10:00
sugarme
056bac2ff8 added CyclicLR 2021-06-10 23:21:11 +10:00
sugarme
ccb7beaf51 changed to SchedulerOptions and added ReduceLROnPlateau 2021-06-10 15:06:00 +10:00
sugarme
aabb8814ac Rewrote scheduler 2021-06-09 20:47:10 +10:00
sugarme
3f17f0b4cb WIP: CosineAnnealingLR 2021-06-08 22:59:25 +10:00
sugarme
be07ef9288 added StepLR and MultiStepLR 2021-06-08 11:46:59 +10:00
sugarme
8e76313b30 updated LambdaLR 2021-06-07 22:34:17 +10:00
sugarme
f26d51d358 added LambdaLR 2021-06-07 22:26: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
720beffa62 fixed nn.Seq Forward nil pointer if layers length = 1 2021-05-15 17:50:08 +10:00
sugarme
d6fb8d88d8 example/jit-train: added 2021-01-02 16:49:36 +11:00
sugarme
82113d7225 added(nn/jit): ForwadT method 2021-01-02 14:24:29 +11:00
sugarme
d0727911c4 feat(tensor/jit): added all 'atm_' API; feat(nn/jit): added TrainableCModule struct 2021-01-02 14:18:49 +11:00
sugarme
c7a8c4b348 changed(tensor/jit): changed to receiver pointer 2021-01-02 12:35:57 +11:00
sugarme
ce7ab51c9b fixed(nn/sequential): fixed missing case number of layers = 1 causing panic 2020-12-11 15:26:54 +11:00
sugarme
9bc9d22409 fixed(nn/varstore): fixed nil pointer at LoadPartial due to not break loop 2020-12-10 09:21:03 +11:00
sugarme
e66046b45e more changes to pointer receiver at nn sub-package and clean-up return variable 2020-11-03 11:56:04 +11:00
sugarme
c38c909977 corrected Int64Option and Float64Option param functions 2020-11-02 22:35:25 +11:00
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