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 |
|
sugarme
|
9ce34e697a
|
WIP(example/yolo)
|
2020-07-14 13:41:18 +10:00 |
|
sugarme
|
f16ab429c9
|
WIP(example/yolo)
|
2020-07-13 22:00:03 +10:00 |
|
sugarme
|
600adf506a
|
feat(tensor): unit test
|
2020-07-11 16:42:27 +10:00 |
|
sugarme
|
94c1af9671
|
fix(tensor/index): fixed incorrect counting of number of NewAxis
|
2020-07-10 11:10:26 +10:00 |
|
sugarme
|
8107d429b6
|
fix(nn/module): ForwardT; feat(example/cifar)
|
2020-07-08 15:19:40 +10:00 |
|
sugarme
|
0a7f3afe6e
|
feat(vision/mobilenet)
|
2020-07-07 11:53:52 +10:00 |
|
sugarme
|
36c0bac2df
|
feat(vision/densenet)
|
2020-07-06 18:32:43 +10:00 |
|
sugarme
|
3dde7e76ad
|
WIP(vision/efficientnet)
|
2020-07-06 12:02:38 +10:00 |
|
sugarme
|
76f4b41ad1
|
fix(nn/varstore): fixed incorrect Load function and replaced '|' with '.'
|
2020-07-02 12:33:18 +10:00 |
|
sugarme
|
32d4a68e12
|
WIP(nn/varstore_test)
|
2020-07-02 09:05:08 +10:00 |
|
sugarme
|
fcbc4ca870
|
feat(example/jit)
|
2020-06-30 20:01:01 +10:00 |
|
sugarme
|
2b88cf2a87
|
WIP(vision/resnet): added more func
|
2020-06-29 17:56:11 +10:00 |
|
sugarme
|
006fb6bb62
|
feat(vision/imagenet): added
|
2020-06-29 15:42:36 +10:00 |
|
sugarme
|
13c94d5960
|
feat(tensor/jit): add wrapper functions
|
2020-06-27 20:32:09 +10:00 |
|
sugarme
|
c6f06d208a
|
feat(tensor/jit): completed
|
2020-06-27 17:11:17 +10:00 |
|
sugarme
|
a8f105d0ae
|
WIP(tensor/jit): completed IValue
|
2020-06-27 14:25:36 +10:00 |
|