lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,534 Commits 57 Branches 331 Tags 26 MiB
Commit Graph

185 Commits

Author SHA1 Message Date
Dun Liang 1f06bbf22e fix dropout2d 2022-11-13 16:36:45 +08:00
Dun Liang 74932f3c32 add dropout 2d 2022-11-13 16:17:48 +08:00
lzhengning b2fb32aa52 feats: defaults to keep the first dim in nn.Flatten 2022-09-29 22:14:29 +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
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
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
Dun Liang 51037cbe9f polish keepdim and keepdims 2022-08-13 14:15:33 +08:00
lzhengning 75b77e411f feature: official ROCm support 2022-08-06 00:44:33 +08:00
Dun Liang 8b1d620b22 Merge branch 'master' of github.com:Jittor/jittor 2022-05-26 14:58:33 +08:00
Dun Liang 31fbabfb6f add th_mode 2022-05-25 17:13:43 +08:00
Dun Liang ba266fa99c polish log_softmax precision 2022-05-18 15:37:09 +08:00
li-xl 9b7a184808 polish cross_entropy_loss 2022-05-13 20:04:50 +08:00
lzhengning cac2b83a1c update copyright to 2022 2022-04-22 18:01:02 +08:00
Dun Liang 0666456a2f v 1.3.3 memory optimization 2022-04-22 15:04:06 +08:00
cxjyxx_me 45df1a4eee fft polish 2022-03-30 01:29:25 -04:00
cxjyxx_me b642b8f1d1 Merge branch 'master' into fft 2022-03-26 23:30:35 -04:00
cxjyxx_me b04ad0ccb4 Merge branch 'master' into fft 2022-03-26 23:22:55 -04:00
Dun Liang 438c8ba692 Merge branch 'master' of github.com:Jittor/jittor 2022-03-15 21:18:08 +08:00
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
Dun Liang 81b847e6f0 optimize softmax 2022-02-18 16:45:08 +08:00
yang guo ye e9aca0444c
Update nn.py 
fix typo
 2022-01-30 11:59:38 +08:00
cxjyxx_me e9f681de53 fft 2022-01-25 19:32:06 +08:00
Dun Liang 85ba4898cf add function version conv transpose 2022-01-04 17:29:58 +08:00
Dun Liang ac66897047 polish conv transpose group 2021-12-17 17:54:41 +08:00
Dun Liang 6bc17cb99c polish interpolate 2021-12-06 12:21:18 +08:00
lzhengning 205bd011e7 polish embedding 2021-10-28 13:44:08 +08:00
lzhengning ca2b4f20d0 polish doc for conv, rnn, and activation 2021-10-11 11:54:02 +08:00
lzhengning 9ec300f2aa Polish RNN, GRU, and LSTM 
1. Use Cudnn to speed rnn
2. Fix: document string not correctly rendered
3. Fix: RNN cannot accept relu nonlinearity
4. feat: add default hidden state when executing rnn/gru/lstm
 2021-09-28 17:54:56 +08:00
Dun Liang 4fb462e1b9 polish sub module save load 2021-09-13 11:03:28 +08:00
Dun Liang f807a28e6b add ctcloss 2021-09-08 17:45:17 +08:00
Dun Liang 057dd95658 fix typo 2021-09-08 13:46:51 +08:00
Dun Liang 39165cccfb polish bmm 2021-08-04 19:36:02 +08:00
Dun Liang d482161be0 floor keep type and better mem_info 2021-07-30 20:40:21 +08:00
Dun Liang 0cf77ea11c add bilinear module 2021-07-23 23:31:21 +08:00
cxjyxx_me d89f1419f9 Merge branch 'master' of github.com:Jittor/jittor 2021-07-20 14:16:49 +08:00
cxjyxx_me b65286072e add TODO 2021-07-20 14:16:22 +08:00
Dun Liang 9c03fd4f75 polish conv3d bias 2021-07-17 16:57:18 +08:00
Dun Liang ee5900bd90 add functional normalize 2021-07-16 18:18:02 +08:00
Dun Liang 6d4437043f polish cross_entropy_loss 2021-07-15 15:05:07 +08:00
uyzhang 304d5034fd add reduction parameter to cross_entropy_loss 2021-07-14 21:42:21 +08:00
Dun Liang 39981b896e polish prelu param 2021-07-09 15:27:44 +08:00
cxjyxx_me 82eca30ee6 tf_resize 2021-07-06 15:31:20 +08:00
Dun Liang fe11a83155 resize nearest polish 2021-06-25 23:44:50 +08:00
Dun Liang 1246c37692 fix ce 2021-06-24 21:39:43 +08:00
lzhengning 4e1ee052f7 cross entropy fix & feat 
fix: wrong cross entropy when using ignore_index
feat: support weight of labels
 2021-06-24 16:50:39 +08:00
Dun Liang 8aa478fa5e add fake Parameter and backward interface 2021-06-23 21:45:11 +08:00
Dun Liang 698fc6fe88 add parameter list and dict 2021-06-22 21:25:38 +08:00