lzhengning
|
db81cc938f
|
fix: default to use CUDA when both CUDA and ROCm are installed
|
2022-10-17 12:09:36 +08:00 |
Dun Liang
|
335a2e5c1d
|
support conv
|
2022-10-13 01:20:14 +08:00 |
Dun Liang
|
042c3610a3
|
improve cuda restart issue
|
2022-10-10 17:31:53 +08:00 |
Dun Liang
|
c12549020f
|
Merge branch 'master' of github.com:Jittor/jittor
|
2022-10-10 17:30:36 +08:00 |
Dun Liang
|
524564b763
|
fix some acl issue
|
2022-10-07 14:20:18 +08:00 |
Zheng-Ning Liu
|
7827c45047
|
Merge pull request #378 from Exusial/npth
Polish loading weights for PyTorch .pth files
|
2022-10-06 16:50:35 +08:00 |
Exusial
|
17db747ae4
|
merge.
|
2022-10-06 16:42:08 +08:00 |
Dun Liang
|
a57de764f6
|
fix issue #401, improve atan and atan2
|
2022-10-05 17:21:51 +08:00 |
Dun Liang
|
9a5e7ea6f5
|
code op support more than 10 args
|
2022-10-05 16:43:00 +08:00 |
Exusial
|
b21ee40360
|
fix wrong transpose.
|
2022-10-03 18:38:14 +08:00 |
lzhengning
|
b2fb32aa52
|
feats: defaults to keep the first dim in nn.Flatten
|
2022-09-29 22:14:29 +08:00 |
lidongyang
|
601101ea44
|
add new inplace function document
|
2022-09-26 14:35:45 +08:00 |
lidongyang
|
59e49b064d
|
fix trunc_normal_ & add some inplace function
|
2022-09-26 14:35:45 +08:00 |
Zheng-Ning Liu
|
7164b1cc0f
|
Polish document
|
2022-09-22 00:56:15 +08:00 |
lzhengning
|
a44c016408
|
Add a flag
|
2022-09-20 16:23:23 +08:00 |
lzhengning
|
1a91f1dd01
|
update version to 1.3.5.15:
1. pretty-print jt.Var
2. support area mode in interpolate
3. support reshape empty var with uncertain dimension
|
2022-09-19 21:46:53 +08:00 |
lzhengning
|
e050dd6dc9
|
pretty print jt.Var
|
2022-09-19 21:44:48 +08:00 |
Exusial
|
b258cf3a84
|
add documentation and tests.
|
2022-09-18 23:41:57 +08:00 |
Exusial
|
56255578f9
|
Add interpolate area support.
|
2022-09-18 23:41:57 +08:00 |
lzhengning
|
05ed6c7e34
|
fix: support reshape empty var with uncertain dimension
|
2022-09-16 15:36:33 +08:00 |
Dun Liang
|
e661f19e20
|
Merge branch 'master' of github.com:Jittor/jittor
|
2022-09-16 01:01:20 +08:00 |
Dun Liang
|
c0ed98cbd6
|
polish mem leak problem
|
2022-09-16 00:59:29 +08:00 |
uyzhang
|
e0bd748ff1
|
fix droppath
|
2022-09-15 16:08:02 +08:00 |
uyzhang
|
658ab32bda
|
add droppath in jt.nn
|
2022-09-14 16:35:38 +08:00 |
Exusial
|
48997cd2b3
|
fix linspace zero division.
|
2022-09-14 16:14:36 +08:00 |
lzhengning
|
46e2dcbefd
|
1. polish index_op doc; 2. polish pyi files
|
2022-09-14 16:09:14 +08:00 |
Xiang-Li Li
|
95db8b310e
|
Merge pull request #389 from Jittor/kldiv
feat: add KLDivLoss in jittor.nn
|
2022-09-13 21:30:47 +08:00 |
lzhengning
|
9d176643a0
|
feat: add KLDivLoss in jittor.nn
|
2022-09-13 21:22:51 +08:00 |
lzhengning
|
3cce690218
|
feat: add one_hot in jt.nn
|
2022-09-07 21:48:53 +08:00 |
root
|
2c8279ebe8
|
update installation in README
|
2022-09-07 15:32:40 +08:00 |
lixl19
|
83622edc97
|
polish unique
|
2022-08-30 20:17:30 +08:00 |
lixl19
|
c3efd2bd9d
|
fix unique
|
2022-08-30 19:31:45 +08:00 |
Xiang-Li Li
|
b54ce7f51b
|
Merge pull request #374 from zjp-shadow/zjp
Update the unique (add new features)
|
2022-08-30 19:20:26 +08:00 |
root
|
87f6c4296e
|
Update compilation and Add the test of cuda
|
2022-08-30 16:28:08 +08:00 |
Exusial
|
5eaccf538d
|
add epillsis (numpy version) support.
|
2022-08-29 14:09:16 +08:00 |
Exusial
|
6339f62f56
|
fix einsum space interpret error.
|
2022-08-27 22:25:00 +08:00 |
Exusial
|
b6c9421a9a
|
fix multiple read.
|
2022-08-27 11:11:22 +08:00 |
Exusial
|
e58e0d2247
|
fix index.
|
2022-08-27 00:03:34 +08:00 |
Zheng-Ning Liu
|
1025d94ffa
|
Update version
|
2022-08-25 00:25:21 +08:00 |
Exusial
|
880441f193
|
allclose, add log.
|
2022-08-25 00:23:28 +08:00 |
Exusial
|
125adc7a04
|
add support for non-zip pytorch pth format.
|
2022-08-25 00:23:28 +08:00 |
root
|
5ffa2747a8
|
update the unique
|
2022-08-23 01:53:51 +08:00 |
Exusial
|
3b814405ce
|
allclose, add log.
|
2022-08-19 21:57:28 +08:00 |
Exusial
|
92b87448a2
|
add support for non-zip pytorch pth format.
|
2022-08-18 14:47:34 +08:00 |
Zheng-Ning Liu
|
89bf097445
|
Update version
|
2022-08-17 13:54:59 +08:00 |
Zheng-Ning Liu
|
ae5988f512
|
Update cutt_transpose_op.cc
|
2022-08-17 13:51:55 +08:00 |
lzhengning
|
72eb7c76bd
|
fix: cutt tranpose error when input is empty
|
2022-08-17 13:51:55 +08:00 |
Dun Liang
|
d2b5c281b2
|
polish self update issue
|
2022-08-17 13:51:15 +08:00 |
Dun Liang
|
9e599ebf9d
|
polish win bug
|
2022-08-13 23:41:28 +08:00 |
Dun Liang
|
64a30cc6fe
|
add reuse array optimization
|
2022-08-13 23:26:31 +08:00 |