Commit Graph

117 Commits

Author SHA1 Message Date
Goncalves Henriques, Andre (UG - Computer Science)
d21543e88c Try to fix retain 2024-04-21 17:37:41 +01:00
Goncalves Henriques, Andre (UG - Computer Science)
cc72639158 Try to fix retain 2024-04-21 17:24:12 +01:00
Goncalves Henriques, Andre (UG - Computer Science)
22e75becf0 Try to add retain grad 2024-04-21 15:30:33 +01:00
sugarme
983939a9b8 clean up cgo flags 2023-11-29 20:55:15 +11:00
sugarme
a1c9ac1dc5 fixed incorrect cbool conversion in autogenerated APIs 2023-10-31 00:02:08 +11:00
sugarme
bf24e57df9 upgrade libtorch2.1.0 2023-10-11 14:53:35 +11:00
sugarme
e9278816b2 fixed incorrect APIs generation 2023-08-05 13:02:56 +10:00
sugarme
aa23a1e59b WIP: pytorch v2 2023-07-26 21:43:21 +10:00
sugarme
1fc13ec55e added tensor IsContiguous and IsMkldnn APIs 2023-07-23 14:08:04 +10:00
sugarme
66599380d4 clean-up 2023-07-23 12:36:55 +10:00
sugarme
dbbb4b97c1 added get/set cuda device for qt testing 2023-07-11 15:35:36 +10:00
sugarme
c1ee7689ad fixed check null at tensor ops return slice of tensors and clean-up 2023-07-07 16:08:15 +10:00
sugarme
e29288f63e fetched from master 2023-07-04 23:26:20 +10:00
Ryan
19a1b2f028 Use more accurate type cast when calling into C 2022-11-24 19:02:53 -06:00
sugarme
6397c08813 fixed jit test and regenerated APIs 2022-03-16 12:34:38 +11:00
sugarme
c24abefcf3 upgraded to libtorch-1.11 required cuda11.3 2022-03-13 12:56:11 +11:00
sugarme
66cea05f28 fixed #67 2022-02-19 16:59:33 +11:00
sugarme
6e07f2cca1 added new API: ConstantPadNdWithVal 2022-01-17 21:41:16 +11:00
sugarme
6c4ce7f55f upgrade to libtorch 1.10 2021-11-06 14:44:27 +11:00
sugarme
6c38d54cec deleted dummy_cuda and bump version 2021-07-26 18:00:24 +10:00
sugarme
42fe023433 fixed cuda err: undefined reference to at::cuda::warp_size() 2021-07-24 17:33:17 +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
77031295cb temp fixed huge lr groups returned from C 2021-07-08 15:38:00 +10:00
sugarme
ec77385d31 flip back fixed 44 as not working on linux 2021-07-06 19:11:32 +10:00
sugarme
af8f999635 fixed double free in vision/image resizePreserveAspectRatio and wrong ctype at libtch 2021-07-06 18:58:15 +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
acdf396796 fixed #30 - save model crashed randomly due to segmentfault 2021-05-11 15:52:51 +10: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
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
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
4a129c0c4b travis CI 2020-10-12 21:43:10 +11:00
sugarme
41f2d149b7 fixed travisCI 2020-10-12 21:30:40 +11:00
sugarme
5a4a65e35c updated Travis CI 2020-10-12 21:21:37 +11:00
sugarme
299793804a added fixed lib cgo flags 2020-10-12 17:03:12 +11:00
sugarme
aee81d22b3 update scripts for GPU and CPU 2020-10-11 08:45:49 +11:00
sugarme
d55436e1ea update GPU script 2020-10-10 21:17:58 +11:00
sugarme
e5735c77dc WIP(patch): AtgSplit func 2020-09-24 09:52:31 +10:00
sugarme
3f95b46f6c clean-up lib.go 2020-07-30 15:43:46 +10:00
sugarme
57e963cfe5 BREAKING CHANGE: make master work for travis and cpu branch 2020-07-26 10:30:20 +10:00
sugarme
0ff2f910f2 BREAKING CHANGE: switch to auto-generated 2020-07-22 15:56:30 +10:00
sugarme
959a1c8a99 change(tensor): move ts.AtDevice from c_generated to tensor.go 2020-07-21 16:23:44 +10:00
sugarme
530acf0894 WIP(example/yolo) 2020-07-15 13:42:40 +10:00