Commit Graph

144 Commits

Author SHA1 Message Date
sugarme
47c6c60561 fixed tensor.OfSlice() and removed TravisCI config for now 2022-02-24 10:47:15 +11:00
sugarme
2d5031009b added tensor.FromCtensor() 2022-02-16 11:39:27 +11:00
sugarme
8d586824c2 fixed tensor.ValueGo() return []int 2022-02-13 23:50:45 +11:00
sugarme
961080760f fixed ts.OfSlice() not supporting []int data type 2022-02-13 22:46:50 +11:00
sugarme
6e07f2cca1 added new API: ConstantPadNdWithVal 2022-01-17 21:41:16 +11:00
sugarme
b1fa1004e0 clean up 2022-01-17 17:20:40 +11:00
sugarme
45ff5b2e5b fixed wrong tensor method 'Meshgrid' 2022-01-17 17:18:54 +11:00
sugarme
6c4ce7f55f upgrade to libtorch 1.10 2021-11-06 14:44:27 +11:00
sugarme
0a846f61ca fixed #58 2021-09-15 10:33:57 +10:00
sugarme
4060f00193 added missing Tensor methods returning multiple tensor values 2021-08-15 21:59:10 +10:00
sugarme
5d8751bfaf updated APIs changes 2021-07-23 00:54:41 +10:00
sugarme
49bb517da3 generated new APIs 2021-07-22 19:00:53 +10:00
sugarme
bc12fc8605 WIP: update to v1.9 2021-07-22 00:38:55 +10:00
sugarme
4fd5c5059d export AugmentOption and correct ColorJitter output dtype 2021-07-20 13:58:45 +10:00
sugarme
c89e4b3ba1 reworked on resnet, densenet, added BCELoss, CrossEntropyLoss, changed DataLoader.Reset() 2021-07-14 10:38:11 +10:00
sugarme
b243c69a9d fixed delete tensor inside function at tensor.CrossEntropyForLogits and tensor.AccuracyForLogits 2021-06-19 09:46:24 +10:00
sugarme
834a0ec9cb Merge branch 'master' of github.com:sugarme/gotch into scheduler 2021-06-12 18:47:05 +10:00
sugarme
6f8ec3b69f added AdamW optimizer 2021-06-12 18:40:42 +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
3ba319edc4 attempt to fix #38 2021-06-03 15:04:38 +10:00
sugarme
7292c3575e added image augmentation and minor fixed on ts.Lstsq 2021-05-22 21:02:22 +10:00
sugarme
367f57fdea fixed lstsq API 2021-05-19 13:59:21 +10:00
sugarme
0d86ab1cf3 added missing ts.Lstsq API 2021-05-19 13:48:29 +10:00
sugarme
a794d69b2c fix(tensor/jit): fixed IValue.Value() returns []interface{} instead of
[]Tensor.
2021-02-06 09:44:50 +11:00
sugarme
d6fb8d88d8 example/jit-train: added 2021-01-02 16:49:36 +11:00
sugarme
86c817b94d jit/train, eval: fixed missing param 2021-01-02 15:45:20 +11:00
sugarme
2fb8acef07 added train and eval atm_ API 2021-01-02 15:40:52 +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
9bc9d22409 fixed(nn/varstore): fixed nil pointer at LoadPartial due to not break loop 2020-12-10 09:21:03 +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
94197e0710 WIP: npy.go 2020-11-16 23:37:44 +11:00
sugarme
9a4646f331 WIP: npy.go 2020-11-16 17:38:16 +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
sugarme
5d60adf0fd tensor/print: added basic tensor print using fmt Formatter interface 2020-11-09 13:35:28 +11:00
sugarme
ca8e51fad8 switched back to lib.ato_add_parameters_old as param group not updated yet 2020-11-03 00:14:56 +11:00
sugarme
c38c909977 corrected Int64Option and Float64Option param functions 2020-11-02 22:35:25 +11:00
sugarme
b2bc72b1d4 WIP: pytorch v1.7.0 2020-11-02 18:54:22 +11:00
sugarme
a6d09580aa corrected pointer receiver conversion in vision sub-packages and examples 2020-11-01 11:59:08 +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
d93cf1b996 reverse changed #10 2020-10-27 17:30:58 +11:00
sugarme
1914aac74e tensor/Drop: updated comment 2020-10-24 09:36:06 +11:00
sugarme
95db45896e feat(tensor): ts.MustDrop() now can call multiple times 2020-10-24 00:29:23 +11:00
sugarme
34efa9e3da feat(patch): completed 9 missing APIs for 'tensor *atg_' pattern 2020-09-24 11:03:21 +10:00
sugarme
e5735c77dc WIP(patch): AtgSplit func 2020-09-24 09:52:31 +10:00