Commit Graph

167 Commits

Author SHA1 Message Date
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
Dun Liang 826513a156 add safe_clip and safe_log 2021-06-11 14:11:58 +08:00
Dun Liang fd798d9925 conv3d optimization 2021-06-10 15:27:12 +08:00
Dun Liang b6fe53e984 optimize conv3d 2021-06-10 15:11:03 +08:00
Dun Liang 88483fedbc add layernorm3d 2021-06-04 14:07:07 +08:00
Dun Liang 6eafa4dd66 Merge branch 'master' of https://github.com/jwzxgy2007/jittor 2021-06-04 14:01:11 +08:00
Dun Liang 2efd7be357 add UpsamplingBilinear2d, AdaptiveMaxPool2d 2021-05-11 16:29:59 +08:00
Dun Liang 862d564d85 add distribution for rl 2021-04-29 23:04:02 +08:00
li-xl b75b6def89 add rnncell and grucell 2021-04-29 16:08:05 +08:00
lzhengning b0669c11c0 feats: polish rnnbase 2021-04-29 15:05:39 +08:00
lzhengning 026dfb8fa2 feat: add rnn, gru 2021-04-29 14:58:24 +08:00
lzhengning 18923fcf3d feat: bidirectional 2021-04-29 13:32:18 +08:00
lzhengning 5f6c170121 feat: add lstm 2021-04-28 15:11:08 +08:00
li-xl 8f7c997afb update upsample 2021-03-12 15:34:34 +08:00
li-xl 9b67168f18 update zeropad 2021-03-11 16:08:52 +08:00
Dun Liang 36e7acfc1f polish auto_parallel and upsample 2021-03-11 16:05:34 +08:00
Dun Liang 8616c01a74 polish conv1d bias 2021-03-10 15:25:55 +08:00
Dun Liang 40ce9e605a polish conv1d weight 2021-03-10 13:35:06 +08:00
Exusial 0d7f0db1b9
add unfold,fold,bicubic,qr,eye (#180)
* fix bicubic,add fold.

* add eye.

* add test for qr,bicubic,fold,unfold.

* fix bicubic and fold to code_op ver.add grad test.

* add docs.update pinv to support (..,M,N) shape

* edit maintainer and testfunc's name.

* fix nn

* fix nn

Co-authored-by: Exusial <2247838039@qq.com>
Co-authored-by: Gword <471184555@qq.com>
2021-03-02 16:38:28 +08:00