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
Goncalves Henriques, Andre (UG - Computer Science)
9257404edd
Move the name of the module
2024-04-21 15:15:00 +01:00
sugarme
db661550e5
Merge pull request #120 from sugarme/cflags
...
clean up cgo flags
2023-11-29 20:58:54 +11:00
sugarme
983939a9b8
clean up cgo flags
2023-11-29 20:55:15 +11:00
sugarme
2a3a963aef
minor release
2023-10-31 00:12:06 +11:00
sugarme
0454fa93b5
Merge pull request #118 from sugarme/api
...
fixed incorrect cbool conversion in autogenerated APIs
2023-10-31 00:03:38 +11:00
sugarme
a1c9ac1dc5
fixed incorrect cbool conversion in autogenerated APIs
2023-10-31 00:02:08 +11:00
sugarme
05eb7f538b
Merge pull request #113 from pjongy/fix_concurrency_bug
...
Fix concurrency bug in generating tensor name
2023-10-24 14:06:49 +11:00
pjongy
0e95225655
Fix concurrency bug in generating tensor name
...
TensorCount should be wrapped by lock
- Tensor name came from {PREFIX}_{CURRENT_TENSOR_COUNT}
- atomic Addint64 is good but generating name from this is quite bug
prone
2023-10-23 13:46:08 +00:00
sugarme
e0fce090d5
Merge pull request #112 from pjongy/minor_cleanup_memory
...
MINOR: Move runtime.SetFinalizer into freeXXX
2023-10-19 16:10:39 +11:00
sugarme
96ed6cb348
Merge pull request #111 from pjongy/bugfix_110_concurrency
...
Fix concurrency issue in newTensor
2023-10-19 16:06:08 +11:00
pjongy
6f7d68572b
Fix build error
2023-10-19 04:18:26 +00:00
pjongy
8031bb85f5
MINOR: Move runtime.SetFinalizer into freeXXX
2023-10-19 04:15:16 +00:00
pjongy
f2b5c911d3
Fix same but in scalar
2023-10-19 04:10:42 +00:00
pjongy
7533f43c2e
Fix concurrency issue in newTensor
2023-10-19 03:17:21 +00:00
sugarme
cd0565716f
Merge branch 'master' of ssh://github.com/sugarme/gotch
2023-10-11 15:46:44 +11:00
sugarme
293bad50ab
fixed lib to C++17
2023-10-11 15:46:11 +11:00
sugarme
8a97ca1559
Merge pull request #106 from sugarme/libtorch2.1
...
upgrade libtorch2.1.0
2023-10-11 14:54:35 +11:00
sugarme
bf24e57df9
upgrade libtorch2.1.0
2023-10-11 14:53:35 +11:00
sugarme
f06e858def
clean-up broken badges
2023-10-11 12:34:56 +11:00
sugarme
f2c1321cb0
upgrade Go version 1.20
2023-10-11 12:32:20 +11:00
sugarme
52d0bc412b
update installation script
2023-10-11 12:31:45 +11:00
sugarme
4f03dec060
upgrade libtorch v2.0
2023-10-11 12:00:02 +11:00
sugarme
511dd161b0
updated for versioning
2023-10-11 11:56:28 +11:00
sugarme
83394ef093
Merge pull request #105 from sugarme/libtorch2
...
Libtorch2
2023-10-11 11:30:26 +11:00
sugarme
f0b87eb001
fixed ununique tensor name caused double free tensor error
2023-08-31 22:42:07 +10:00
sugarme
1cffab577c
fixed Linear.Forward with bias is nil and clean up
2023-08-12 22:09:34 +10:00
sugarme
b3d821d34e
temp use of ts.Randn() instead of ts.Uniform() as it causes mem leak
2023-08-12 20:42:39 +10:00
sugarme
163e625426
fixed mem blow-up due to gradient accumulation when loading model pretrained
2023-08-12 15:46:51 +10:00
sugarme
ef00723027
remove linear.bias init when not required
2023-08-09 16:25:09 +10:00
sugarme
e9278816b2
fixed incorrect APIs generation
2023-08-05 13:02:56 +10:00
sugarme
bdf252d831
added debug stack trace to tensor error
2023-07-28 18:06:53 +10:00
sugarme
3cd8d8560f
generated newTensor() for GC collection
2023-07-26 23:19:38 +10:00
sugarme
aa23a1e59b
WIP: pytorch v2
2023-07-26 21:43:21 +10:00
sugarme
3959bc3f93
fixed read npy failed with single element tensor with zero shape
2023-07-25 13:58:19 +10:00
sugarme
30068c6b41
added ts
2023-07-23 14:54:02 +10:00
sugarme
1fc13ec55e
added tensor IsContiguous and IsMkldnn APIs
2023-07-23 14:08:04 +10:00
sugarme
6aa57478f5
added
2023-07-23 13:10:24 +10:00
sugarme
66599380d4
clean-up
2023-07-23 12:36:55 +10:00
sugarme
d1b9267c77
added missing
2023-07-23 12:17:27 +10:00
sugarme
dbbb4b97c1
added get/set cuda device for qt testing
2023-07-11 15:35:36 +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
c1ee7689ad
fixed check null at tensor ops return slice of tensors and clean-up
2023-07-07 16:08:15 +10:00
sugarme
f9cb2f5cc6
Clean up
2023-07-07 13:20:51 +10:00
sugarme
34e87b1302
added dtype option to nn package
2023-07-07 12:24:56 +10:00
sugarme
523061eca6
reworked gotch.dtype with more dtypes
2023-07-07 00:01:23 +10:00
sugarme
640af9d2df
upgrade golang.org/x/image@v0.5.0
2023-07-06 00:26:39 +10:00