Commit Graph

1099 Commits

Author SHA1 Message Date
Dun Liang b7fff3072a Merge branch 'master' of https://github.com/Jittor/jittor 2021-07-30 20:40:31 +08:00
Dun Liang d482161be0 floor keep type and better mem_info 2021-07-30 20:40:21 +08:00
Zheng-Ning Liu 43b150a6c8
Update auto_diff.py
fixed: symbol not found when running "from jittor_utils import auto_diff; import jittor as jt"
2021-07-30 15:52:52 +08:00
li-xl 4d9d2d8601 Merge branch 'master' of github.com:jittor/jittor 2021-07-29 21:47:51 +08:00
li-xl abccb16248 polish xavier_uniform 2021-07-29 21:47:41 +08:00
Dun Liang 08eeb67de5 fix g++-10 line error 2021-07-29 20:02:14 +08:00
Dun Liang d563826b2b add cuda aarch64 support 2021-07-28 11:41:13 +08:00
Dun Liang 0748fc1854 polish cutt transpose 2021-07-27 11:13:39 +08:00
Dun Liang 01974db52d add exp lr and remove warning as error 2021-07-26 21:21:20 +08:00
Dun Liang ee020b60f7 data gz and register_hook and requires_grads_ 2021-07-26 20:58:15 +08:00
Dun Liang 0cf77ea11c add bilinear module 2021-07-23 23:31:21 +08:00
Dun Liang 7052251098 polish slice ellipsis_with_none 2021-07-23 14:38:45 +08:00
Dun Liang 51a28574c8 add jt_sync env 2021-07-23 14:17:29 +08:00
Dun Liang 31c56a1017 add var_dataset 2021-07-23 13:36:58 +08:00
Dun Liang b0a8943404 add TensorDataset 2021-07-23 13:30:16 +08:00
Dun Liang 87e639730c update version 2021-07-22 22:15:53 +08:00
Dun Liang e3181e706f fix console error 2021-07-22 22:14:51 +08:00
Dun Liang 399060e08c ssl polish 2021-07-21 17:08:25 +08:00
Dun Liang 08fcf01d62 add sw_64 support 2021-07-21 17:04:48 +08:00
Dun Liang c82db520b1 Merge branch 'master' of https://github.com/Jittor/jittor 2021-07-20 21:17:33 +08:00
Dun Liang 69979f71e4 polish setitem optimize 2021-07-20 21:14:55 +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
cxjyxx_me 4e80c2118d adamw 2021-07-20 14:16:11 +08:00
Dun Liang e1472a7a8f polish clip_grad_norm update queue leak 2021-07-19 16:40:02 +08:00
Dun Liang cbc5a98bcf Merge branch 'setitem_inplace' of https://github.com/Jittor/jittor 2021-07-19 16:30:54 +08:00
Dun Liang cf6299009e polish auto diff with backward name 2021-07-19 16:30:05 +08:00
zhouwy19 b5e2780ba7 polish setitem data inplace 2021-07-19 16:27:20 +08:00
Dun Liang 62bbdcd7d9 Merge branch 'auto_diff' of https://github.com/lzhengning/jittor 2021-07-19 11:22:00 +08:00
Dun Liang 150604a1a2 polish mod float negtive 2021-07-18 20:14:41 +08:00
Dun Liang 9bf604934e gopt_disable && JT_macro fix && mod grad 2021-07-18 17:18:45 +08:00
Dun Liang 242084349f add more test for setitem 2021-07-17 19:55:53 +08:00
Dun Liang b4cb572c90 polish setitem inplace opt 2021-07-17 19:48:20 +08:00
Dun Liang 9c03fd4f75 polish conv3d bias 2021-07-17 16:57:18 +08:00
lzhengning b61065e8b3 Polish auto_diff.py
- Use the layer name as file name to be cached.
- Now it can compare input and output of layers with the same name, even if the networks are not identical.
- Add a convenient way to provide identical input data (save_input, load_input).
2021-07-17 13:58:31 +08:00
Zheng-Ning Liu e371a9d887
Merge pull request #240 from lzhengning/doc
doc: set seed
2021-07-17 13:34:06 +08:00
Dun Liang ee5900bd90 add functional normalize 2021-07-16 18:18:02 +08:00
Dun Liang db9d2a808a fix notebook 2021-07-16 15:47:58 +08:00
Dun Liang 6d4437043f polish cross_entropy_loss 2021-07-15 15:05:07 +08:00
Dun Liang 5a73db73e1 naive opt state_dict 2021-07-14 21:51:52 +08:00
Dun Liang 7ff46370e3 add learn jittor in 60 min 2021-07-14 21:51:52 +08:00
uyzhang 304d5034fd add reduction parameter to cross_entropy_loss 2021-07-14 21:42:21 +08:00
zhouwy19 9491b68031 add StepLR 2021-07-13 15:53:24 +08:00
Dun Liang 39981b896e polish prelu param 2021-07-09 15:27:44 +08:00
cxjyxx_me e7d8d8e2fd Merge branch 'master' of github.com:Jittor/jittor 2021-07-06 15:31:26 +08:00
cxjyxx_me 82eca30ee6 tf_resize 2021-07-06 15:31:20 +08:00
Dun Liang 2db62c17a6 polish pool interface 2021-07-05 21:20:10 +08:00
Dun Liang 9f58355341 add no_fuse and polish dataset 2021-07-05 21:17:57 +08:00
Dun Liang 8cab7d2a3e support different stride and kernel for unpool 2021-07-05 17:11:43 +08:00
Dun Liang f00ffa3861 add unpool doc 2021-07-05 10:58:01 +08:00