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

233 Commits

Author SHA1 Message Date
lidongyang 393684f196 polish nn.Sequential attribute 2024-06-05 22:31:20 +08:00
Hanyuxuan f20ea9dcf1 fix illegal parameters of ConvTranspose and Pool,issue #478,#480,#481,#482,#483 2024-05-31 15:23:42 +08:00
Hanyuxuan 26963bc70f fix Pad2d with illegal padding,issue #464,#465,#466,#467 2024-05-31 14:50:47 +08:00
Hanyuxuan 64c6400070 check x.shape and kernel_size of Pool and Pool3d,issue #461,#463 2024-05-31 14:33:07 +08:00
Hanyuxuan c7e31604c2 fix illegal parameters of PixelShuffle of issue #458,fix validity of concat of issue #459 2024-05-30 19:03:17 +08:00
Hanyuxuan 981f60c381 fix illegal parameters of Conv2d issue #471,#472,#473,#474,#475,#476,#477 2024-05-30 16:10:06 +08:00
Hanyuxuan 681174a606 a ValueError fix of issue #450 2024-05-30 14:29:34 +08:00
DongYang Li 4e41e6b070 Merge pull request #539 from fansunqi/issue523_branch 
fix issue 523;update jt.nn.Conv1d/Conv3d/conv2d/conv3d
 2024-05-20 21:42:16 +08:00
范孙奇 dd9ac69eec fix issue 523;update jt.nn.Conv1d/Conv3d/conv2d/conv3d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-17 09:35:12 +00:00
DongYang Li 9943ddf8de Merge pull request #536 from fansunqi/issue528_branch 
fix issue 528;update conv_transpose
 2024-05-16 15:19:19 +08:00
DongYang Li ee3c68ce7a Merge pull request #538 from fansunqi/issue525_branch 
fix issue 525;update jt.nn.Reflection2d/Replication2d
 2024-05-16 14:43:28 +08:00
DongYang Li 136a710775 polish PixelShuffle in nn.py 2024-05-16 14:35:21 +08:00
DongYang Li 75429f83b7 Merge pull request #534 from fansunqi/master 
fix issue 531,530;update jt.nn.PixelShuffle/jt.histc
 2024-05-16 14:27:24 +08:00
范孙奇 82595dc766 fix issue 525;update jt.nn.Reflection2d/Replication2d 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 13:49:26 +00:00
范孙奇 9baccaed4d fix issue 528;update conv_transpose 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 12:21:21 +00:00
范孙奇 ab39f8283c fix issue 531 
Signed-off-by: 范孙奇 <fansq20@mails.tsinghua.edu.cn>
 2024-05-15 11:14:24 +00:00
514flowey 290f2aec60 add complex matmul, inv, qr, eig, and svd 2024-05-08 21:09:07 +08:00
lidongyang 4ae9171578 fix merge conflict 2023-12-28 16:10:57 +08:00
Dun Liang fa40284b60 Merge branch 'master' of https://github.com/Jittor/jittor 2023-12-27 23:18:17 +08:00
Dun Liang e260f2b6bd fitten code 2023-12-05 01:34:26 +08:00
lidongyang 1f30881c6e polish GroupNorm&fix different op 2023-12-04 15:46:03 +08:00
Dun Liang c485fdc07b madd bf16 2023-09-08 22:43:31 +08:00
yang guo ye 21b551411e
Merge pull request #486 from 514flowey/complex_number 
add complex matmul
 2023-08-06 17:05:52 +08:00
514flowey 14c39a7a53 add complex matmul 2023-08-06 16:58:50 +08:00
Dun Liang 5b6c26ede7 Merge commit '1018e3b2' into HEAD 2023-07-07 22:08:03 +08:00
li-xl 100d1bdd12 support scalar in getitem && fix bugs for jtorch 2023-06-19 16:06:19 +08:00
Dun Liang d595c8b00f Merge branch 'master' of github.com:Jittor/jittor 2023-06-10 01:01:30 +08:00
Dun Liang 2d4a82d5e5 polish mish activation, thanks! 2023-06-10 00:59:39 +08:00
li-xl b721d851b5 add transformers support 2023-05-24 06:25:25 +08:00
Dun Liang f26fdfea22 LLM training tuning 2023-05-16 23:11:05 +08:00
514flowey 10d9fb1431 add complex number 2023-05-02 18:20:52 +08:00
lzhengning b824ab83c0 update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
Dun Liang 2ea015061c polish swap iter 2023-04-03 12:55:01 +08:00
Dun Liang 00b7783c46 Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 21:14:20 +08:00
Dun Liang e969103399 polish to_float interface 2023-03-31 21:13:59 +08:00
Dun Liang 3300a3e4f3 polish fp16 amp_level3 2023-03-31 21:07:43 +08:00
lzhengning 1119f3cc51 update readme && nn.linear 2023-03-31 19:45:07 +08:00
lzhengning 8a9f0bb904 update load_pytorch.py && add linear 2023-03-31 19:45:07 +08:00
Dun Liang 1e00883ed1 add attr_dict for jittor var 2023-03-24 01:53:30 +08:00
Dun Liang 07f8bd4aff Merge branch 'master' of github.com:Jittor/jittor 2023-03-22 19:47:26 +08:00
Dun Liang db8679748c fp32_guard && ternary broadcast 2023-03-22 19:41:36 +08:00
lzhengning 576b0c9e03 feat: jt.outer & jt.nn.silu 2023-03-21 15:56:21 +08:00
Dun Liang 90850d36ad LLM support with jtorch 2023-03-20 22:33:12 +08:00
Brian Pugh 5dba84ee67 Allow for nn.Sequential to be sliced. 2023-01-15 13:36:29 +08:00
Dun Liang 6fbffb7c62 jtorch compatible 2023-01-06 20:48:04 +08:00
lzhengning 919e863e55 fix depthwise conv error on ROCm 2023-01-04 14:09:53 +08:00
lzhengning d50fe5e754 1. fix compilation failure of fp16_compute.h under rocm platform 
2. add jt.compiler.is_cuda to identify if the current device is CUDA
 2022-12-17 21:36:54 +08:00
Dun Liang 5bc160b19c dataset and dataloader interface polish 2022-11-16 11:31:55 +08:00
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