Commit Graph

109 Commits

Author SHA1 Message Date
li-xl 1f8ea20f01 add elu 2020-12-16 20:03:24 +08:00
Dun Liang c8252ac7fb add conv transpose function 2020-12-13 19:03:43 +08:00
Dun Liang c80e0168c0 Merge branch 'develop' of https://github.com/Jittor/jittor 2020-12-07 22:49:47 +08:00
Dun Liang 514b60dea8 Merge branch 'master' of https://github.com/Jittor/jittor 2020-12-07 22:49:33 +08:00
cxjyxx_me 895f6ecee2 depthwise_conv 2020-12-07 22:43:11 +08:00
cxjyxx_me 14324f364b mrege 2020-12-02 16:57:36 +08:00
Jittor 260a052db9
Update nn.py
fix a simple bug
2020-11-28 20:27:07 +08:00
li-xl 3c7d59ea16 fix Layernorm bugs 2020-11-27 15:19:31 +08:00
Xiang-Li Li 5dfef0af13
Merge branch 'develop' into lxl 2020-11-27 15:11:36 +08:00
Zheng-Ning Liu d1975d9d7e
Add aliases in nn.py 2020-11-27 14:58:30 +08:00
Dun Liang 79579b4a30 merge loop var optimize 2020-11-25 22:15:41 +08:00
li-xl 13941112e1 Merge branch 'master' of https://github.com/jittor/jittor into lxl 2020-11-25 19:23:28 +08:00
Zheng-Ning Liu 798018880e
Update BatchNorm1d
Change default value of affine in BatchNorm1d to be True.
2020-11-25 13:30:21 +08:00
cxjyxx_me 97a99b6a24 depthwise_conv 2020-11-25 11:13:47 +08:00
li-xl 1045f04c4b add misc and bcelogits with pos_weight 2020-11-10 11:34:39 +08:00
li-xl 32af287461 Merge branch 'master' of https://github.com/jittor/jittor 2020-11-08 16:04:16 +08:00
li-xl 98545bede4 add softplus 2020-11-08 16:04:12 +08:00
Dun Liang 805f48369d better trace_py_var 2020-11-06 14:41:38 +08:00
li-xl cb52493db9 Merge branch 'master' of https://github.com/jittor/jittor 2020-11-03 17:06:20 +08:00
li-xl 06eb505eea add bmm_transpose and pad and merge_loop_flags 2020-11-03 17:04:25 +08:00
zhouwy19 b742939605 update sign 2020-10-28 14:07:42 +08:00
zhouwy19 8e93e36f53 add sign & Sequential __iter__ 2020-10-28 11:31:33 +08:00
li-xl aa502deafb optimize where op && fix vary shape infer 2020-10-26 22:03:53 +08:00
li-xl ca7207e027 Merge branch 'master' of https://github.com/jittor/jittor 2020-10-21 11:37:06 +08:00
li-xl bd506460be remove matmul print 2020-10-21 11:36:59 +08:00
Dun Liang ee268c7456 add __iter__ in sequential 2020-10-17 15:42:14 +08:00
Dun Liang 8d64d98a35 jit_prepare for array op 2020-10-12 18:01:14 +08:00
Dun Liang c792b23f48 version 1.2 2020-10-10 16:38:21 +08:00
Dun Liang d6b9e32428 auto diff hook optimizer and rand 2020-09-25 14:14:28 +08:00
Dun Liang e3d1fafa87 fix update queue and conv transpose 2020-09-16 14:27:13 +08:00
Gword 03aefb620a Merge branch 'master' of https://github.com/Jittor/jittor into attention 2020-08-28 20:49:00 +08:00
Gword 7b41ab678f attention op & Sequential with OrderedDict 2020-08-28 20:48:10 +08:00
zhouwy19 e49a1cefc9 fix ConvTranspose weight init bug 2020-08-23 21:13:06 +08:00
zhowuy19 843b983e90 BCELoss add weight 2020-08-14 17:32:42 +08:00
Dun Liang 1ad00d4580 better error control && fix doc && fix free buffer 2020-08-14 13:49:33 +08:00
zhouwy19 d66606a205 change GroupNorm 2020-08-11 16:36:08 +08:00
Dun Liang f296712c37 add cuda bmm test 2020-07-31 21:14:07 +08:00
Dun Liang 84d9434ecf polish matmul 2020-07-31 21:02:01 +08:00
zhouwy19 1e32486f59 update groupnorm 2020-07-31 08:48:08 +08:00
zhouwy19 cad1845e68 update matmul & ConstantPad2d 2020-07-30 23:33:31 +08:00
zhouwy19 90a1422b3c repeat chunk stack flip Softplus GN grid_sample 2020-07-30 23:07:35 +08:00
Meng-Hao f7d893306a update matmul_transpose 2020-07-24 09:40:06 +08:00
Dun Liang c49ebb1327 Merge branch 'gmh' of https://github.com/Jittor/jittor 2020-07-19 13:40:37 +08:00
Dun Liang bec2e710af add changelog 2020-07-15 10:51:48 +08:00
Meng-Hao a4bb6318e1 update bmm 2020-07-12 17:41:46 +08:00
Meng-Hao a859396aa8 update bmm 2020-07-12 17:33:41 +08:00
Meng-Hao 395733f85a update bmm 2020-07-12 17:29:10 +08:00
MenghaoGuo 225a8f4944
Merge pull request #79 from Jittor/gmh
add bmm function
2020-06-26 11:15:14 +08:00
Dun Liang 993f73cb4e fix contrib 2020-06-23 22:48:42 +08:00
Dun Liang 5de2aec717 add update queue 2020-06-20 20:46:30 +08:00