Commit Graph

1128 Commits

Author SHA1 Message Date
Xiang-Li Li 1987728950
Merge pull request #300 from Jittor/jittor_home
add jittor_home
2022-03-22 21:12:39 +08:00
liuruiyang98 5625147244 Update trunc_normal_ with var = var.add(XX) and var = var.clamp(XX) 2022-03-22 20:48:41 +08:00
liuruiyang98 3dd187f34c Update var = var.erfinv() and var = var.multiply(XX) 2022-03-22 20:48:41 +08:00
liuruiyang98 4480351396 Update trunc_normal_ 2022-03-22 20:48:41 +08:00
li-xl aa45703143 add jittor_home 2022-03-22 20:28:43 +08:00
lzhengning 10c51c6823 fix a compatible issue with onednn 2.5 2022-03-22 17:18:31 +08:00
lzhengning de2fceab22 Transitioning from Intel MKL-DNN to oneDNN 2022-03-22 17:10:45 +08:00
Dun Liang 7d9cff9e24 add cgan demo 2022-03-22 16:14:43 +08:00
Dun Liang 6f4e95e55f add simple_cgan demo 2022-03-22 16:14:43 +08:00
SilenceLi 3c8ed89d5e Update __init__.pyi 2022-03-22 15:56:10 +08:00
Dun Liang d4e8092c93 add test_modules 2022-03-22 14:38:50 +08:00
Dun Liang 31fbf4af9c add _modules and _parameters property 2022-03-22 14:37:44 +08:00
Dun Liang 99d6d6b6e1 Merge branch 'master' of github.com:Jittor/jittor 2022-03-21 17:49:24 +08:00
Dun Liang e7bb2545d3 add acl backend 2022-03-21 17:48:50 +08:00
lixl19 1fa89771c4 polish matmul 2022-03-18 16:21:34 +08:00
lixl19 ad57ec890f polish fp16 2022-03-18 12:56:24 +08:00
Dun Liang 5efb222dd3 add half alias 2022-03-15 22:02:38 +08: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 15f162d4f3 cuda11.4 in windows disabled 2022-03-10 17:11:55 +08:00
Dun Liang 56af8b392f update version 2022-03-10 14:02:11 +08:00
Dun 70c462502a polish windows encoding support 2022-03-10 05:56:56 +08:00
Dun Liang 495d78ad20 fix windows encoding bugs 2022-03-09 18:23:42 +08:00
Dun Liang 53b377ee7d fix windows encoding error, thanks liduan for reporting this bug 2022-03-08 16:40:24 +08:00
Dun Liang 61ea95b76f polish windows notebook 2022-03-08 15:14:37 +08:00
Dun Liang f796ad0d7f fix some jupyter related bugs 2022-03-07 17:15:17 +08:00
Zheng-Ning Liu ca2ec40a59
fix: cumsum not available on CPU 2022-03-02 18:05:35 +08:00
Dun Liang 7cf6165a10 profiler polish 2022-02-26 16:14:52 +08:00
lzhengning 9c04a01143 fix: improve compatibility of Windows and python 10 2022-02-26 14:48:53 +08:00
Dun Liang c5acaa2474 polish example 2022-02-24 21:21:04 +08:00
Dun Liang 120057640c customize code op compilation flags 2022-02-24 21:08:29 +08:00
Xiang-Li Li 83e4f53af5
Update contrib.py
update concat docs
2022-02-24 18:18:16 +08:00
Xiang-Li Li 43d3a47964
Update __init__.pyi
polish docs of arg_reduce
2022-02-24 17:11:57 +08:00
Zheng-Ning Liu 0c8b2e3bb9
Update compiler.py
fix-bug: explicitly set log_v when querying cuda to avoid wrong output of jittor_utils.
2022-02-23 21:32:32 +08:00
Xiang-Li Li 4383886feb
Update misc.py
polish cub_cumsum to support ‘range’  item assignment'
2022-02-22 22:34:14 +08:00
Dun Liang 7c7b856dcf fix op stat 2022-02-20 21:27:31 +08:00
Dun Liang 58e8c650eb rename common_op_type 2022-02-20 21:15:56 +08:00
Dun Liang 2dd70a7002 check crlf 2022-02-20 21:06:09 +08:00
Dun Liang bcb57086c3 fp16 support, opbytype interface 2022-02-20 20:34:49 +08:00
Dun Liang 81b847e6f0 optimize softmax 2022-02-18 16:45:08 +08:00
lzhengning 8c0f66c638 polish nvcc_flags 2022-02-18 14:55:00 +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 5d191e6247 add nvcc_path check 2022-01-18 16:32:23 +08:00
Dun Liang 6a372f5f4f polish reindex reduce fuse op 2022-01-12 16:43:39 +08:00
Dun Liang 70ccdb1f17 polish loop var pass 2022-01-11 21:03:37 +08:00
Dun Liang 6d1b5e42bc polish fuser 2022-01-11 17:25:00 +08:00
Dun Liang 5b4576c4dd polish tensorcore of cublas in cuda 10 2022-01-10 14:35:30 +08:00
Dun Liang f36693c797 add bool setitem alias and vector_to_tuple 2022-01-10 12:36:28 +08:00
Dun Liang 0c35e1a29b update version 2022-01-04 17:30:37 +08:00
Dun Liang 8f6928315e Merge branch 'master' of github.com:Jittor/jittor 2022-01-04 17:30:14 +08:00
Dun Liang 85ba4898cf add function version conv transpose 2022-01-04 17:29:58 +08:00
tupig-7 5fd65c72cb fix one bug in win10 jittor\src\misc\cpu_math.cc(47): error C2065: 'M_PI': undeclared identifier 2022-01-04 16:10:29 +08:00
liuruiyang98 ef86da29f1 Add init.trunc_normal_ 2022-01-04 16:09:57 +08:00
lidongyang 5c1164233e polish make_grid 2022-01-04 16:09:28 +08:00
Dun Liang f20fa98e44 add use_tensorcore flags 2021-12-30 20:59:48 +08:00
Dun Liang 7b9113f828 support tensorcore 2021-12-30 16:28:21 +08:00
li-xl 928f7ae5be Merge branch 'master' of github.com:jittor/jittor 2021-12-30 11:05:23 +08:00
Dun Liang d6b8349203 polish reindex fuse 2021-12-29 17:50:20 +08:00
Dun Liang 3a31e32a1a add disable lock flags 2021-12-28 12:59:43 +08:00
Dun Liang 8c44329def update version 2021-12-27 13:40:54 +08:00
Dun Liang 0d371ebbe4 add __setattr__ and __getattr__ 2021-12-27 13:40:19 +08:00
cxjyxx_me e19c9aef7a adapt ptrace missed 2021-12-27 10:52:55 +08:00
Dun Liang ac66897047 polish conv transpose group 2021-12-17 17:54:41 +08:00
Dun Liang 922c0d8246 remove some tests 2021-12-17 10:30:14 +08:00
Dun Liang 1f2b100be0 add mpi in py example 2021-12-16 17:37:16 +08:00
Dun Liang c82f5146c9 catch more out of memory error 2021-12-16 16:28:39 +08:00
Dun Liang 9013c70ddf update where document 2021-12-16 16:20:00 +08:00
Dun Liang bc083360e7 test array reindex fuse 2021-12-15 15:38:49 +08:00
Zheng-Ning Liu 03dd698e65
fix-doc: concat example 2021-12-15 10:20:58 +08:00
Dun Liang b913cd8254 polish atan2 2021-12-09 14:41:41 +08:00
Dun Liang 6bc17cb99c polish interpolate 2021-12-06 12:21:18 +08:00
lzhengning eae0357224 doc: fix the wrong example of arg_reduce_op 2021-11-25 11:03:55 +08:00
Dun Liang 9af14f4f55 polish rnn grad 2021-11-22 13:29:04 +08:00
Dun Liang 1e1fe66a30 update version 2021-11-22 13:17:31 +08:00
Dun Liang 5cebf93e31 polish rnn grad 2021-11-22 13:16:48 +08:00
lzhengning 38694a1b6e fix: wrong grad of varslice like [::-1] 2021-11-21 16:50:45 +08:00
Zheng-Ning Liu 4bd0c4a2f6
fix: to_tensor applied on a jt.Var
Allow jt.transform.to_tensor applied on a jt.Var (do nothing but return the inputs).
2021-11-09 23:00:57 +08:00
Dun Liang b3ee4a54e4 polish cu key 2021-11-04 20:51:58 +08:00
Dun Liang 0274c22afd polish cu key 2021-11-04 20:51:31 +08:00
Dun Liang 5695a2e61f polish windows and erfinv 2021-11-04 20:43:54 +08:00
li-xl d08afa840d Merge branch 'master' of github.com:jittor/jittor 2021-11-02 17:45:34 +08:00
li-xl 40ed259665 support tensorcore mode 2021-11-02 17:45:08 +08:00
Dun Liang a1a82f6b2c update version 2021-11-01 12:20:45 +08:00
Dun Liang 0a2ca63cbc rename wrapper 2021-11-01 12:01:01 +08:00
Dun Liang 08cc8a0451 fix typo 2021-11-01 11:42:38 +08:00
Dun Liang 47c2f65749 polish data migrate 2021-10-28 19:59:54 +08:00
Dun Liang b488cc7c61 add erfinv 2021-10-28 17:23:45 +08:00
Dun Liang 26db9872c4 polish error msg 2021-10-28 14:56:41 +08:00
lzhengning 5257284740 update version 2021-10-28 13:44:44 +08:00
lzhengning 205bd011e7 polish embedding 2021-10-28 13:44:08 +08:00
lzhengning a778458831 update version 2021-10-26 20:20:28 +08:00
lzhengning 3147634775 polish 2021-10-26 20:14:17 +08:00
lzhengning ea497ce80a polish gen_pyi 2021-10-26 20:14:17 +08:00
Dun Liang db3b20d886 polish gen pyi 2021-10-26 20:14:17 +08:00
lzhengning b5331c9026 gen pyi files 2021-10-26 20:14:17 +08:00
Dun Liang 6c980c2146 add performance test tutorial 2021-10-25 16:12:27 +08:00
Dun Liang 856f201d9f polish init doc 2021-10-22 16:11:05 +08:00
Dun Liang 60f45b6adf polish init document 2021-10-22 16:05:07 +08:00
Dun Liang 10e28e2151 update res2net50 url 2021-10-19 20:01:20 +08:00
Dun Liang 537487695a polish download error msg 2021-10-19 17:48:14 +08:00
Dun Liang 849fb796ce polish no device found error 2021-10-19 13:10:04 +08:00
Dun Liang e24dd4f14a transpose fit pt 2021-10-18 16:02:02 +08:00
Dun Liang 34a9dab5fb polish error msg 2021-10-18 15:37:02 +08:00
Dun Liang 6c64a20493 update version 2021-10-17 15:58:36 +08:00
Dun Liang 57b4973ac1 polish error msg 2021-10-17 15:58:36 +08:00
Dun Liang b077d6c185 polish cuda install 2021-10-17 15:56:44 +08:00
Dun Liang f65b5df1a6 polish msvc install 2021-10-17 14:26:11 +08:00
Dun Liang 878cb365ac add dependency for mpi ops 2021-10-15 16:22:56 +08:00
Dun Liang f9f02df3d6 polish reduce doc and pass_manager 2021-10-15 12:20:50 +08:00
Zheng-Ning Liu e6e7423631
Merge pull request #272 from Jittor/meminfo
polish meminfo
2021-10-14 16:51:31 +08:00
li-xl 2b16f80289 polish meminfo 2021-10-14 16:43:28 +08:00
Dun Liang eee405669f fix issue #271 2021-10-14 16:36:04 +08:00
Dun Liang 7cfd216372 update version 2021-10-13 20:16:32 +08:00
Dun Liang c05193408a cutt for seperate cache 2021-10-13 20:15:17 +08:00
Dun Liang 102ffa31a5 better cache key 2021-10-13 19:17:19 +08:00
Dun Liang 030b30727d add pip_publish 2021-10-12 14:29:13 +08:00
Dun Liang e0d046f449 polish install from source 2021-10-12 14:29:13 +08:00
lzhengning ca2b4f20d0 polish doc for conv, rnn, and activation 2021-10-11 11:54:02 +08:00
Dun Liang 9e93a671c7 add clean cache script 2021-10-11 11:13:43 +08:00
Dun Liang aba912eb22 polish error message 2021-10-09 16:46:53 +08:00
Dun Liang 2951214deb polish getitem win support 2021-10-09 16:07:48 +08:00
Dun Liang b02b3192bc polish space with path 2021-10-09 09:36:06 +08:00
Dun Liang 66d9df2f82 polish win path order 2021-10-05 19:59:30 +08:00
Dun Liang ad015f13f5 polish win path handle 2021-10-05 19:18:35 +08:00
Dun Liang 2bf1b89c5e polish mkl and cutt install 2021-10-03 22:42:18 +08:00
Dun Liang 7bc620f274 add cuda arch recommend 2021-10-01 20:31:57 +08:00
Dun Liang 6300b0908f lower arch support for 30 2021-10-01 19:56:47 +08:00
Dun Liang 0fa0584fd3 polish files and add cuda arch hint 2021-10-01 19:41:04 +08:00
Dun Liang a78c3b4f12 Merge branch 'master' of github.com:Jittor/jittor into win_cuda 2021-10-01 19:16:25 +08:00
Dun Liang 2866c64876 update version 2021-10-01 19:15:33 +08:00
Dun Liang 9914e0918b polish msvc crt shared link with -MD 2021-10-01 19:15:27 +08:00
lzhengning 1be676caad Merge branch 'master' of https://github.com/Jittor/jittor 2021-09-30 17:47:57 +08:00
Dun Liang 1ccf45a03d update version 2021-09-30 17:42:58 +08:00
lzhengning e57f0e4879 updated docs 2021-09-30 16:15:32 +08:00
Dun Liang 752a5c62ed polish readme and add test 2021-09-30 15:33:21 +08:00
Dun Liang 2e6c6079cb Merge branch 'win_cuda' of github.com:Jittor/jittor into win_cuda 2021-09-30 15:20:40 +08:00
Dun Liang f43edfb665 polish readme 2021-09-30 15:20:28 +08:00
Dun Liang 28985cfea6 polish event queue 2021-09-30 14:41:40 +08:00
Dun Liang 0136d24d83 Merge branch 'win_cuda' of github.com:Jittor/jittor into win_cuda 2021-09-30 14:11:54 +08:00
lzhengning 6938c25e19 polish mac support for win_cuda 2021-09-30 13:42:51 +08:00
Dun Liang cfbadbbfb0 polish windows 2021-09-29 20:03:33 +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 9293045993 Merge branch 'master' of github.com:Jittor/jittor into win_cuda 2021-09-28 17:23:11 +08:00
Dun Liang 5c2fb6bfce update version 2021-09-28 17:17:20 +08:00
Dun Liang 740b9c8552 polish win_cuda in linux 2021-09-28 17:13:03 +08:00
cxjyxx_me 9ab20a5536 Merge branch 'master' of github.com:Jittor/jittor 2021-09-27 23:21:03 -04:00
cxjyxx_me 36faa8b617 polish shape=1 array fuse 2021-09-27 23:20:20 -04:00
cxjyxx_me 1f51e985d0 polish bool add/substract 2021-09-27 23:19:44 -04:00
Dun Liang ec80112709 polish link problem 2021-09-27 12:12:08 +08:00
Dun Liang c1ee6d9ed3 polish win_cuda on linux 2021-09-26 19:48:22 +08:00
Dun Liang e77f1ea7cb add nvcc exe suffix 2021-09-26 15:57:08 +08:00
Dun Liang 123e915bb3 support cuda win 2021-09-26 15:35:18 +08:00
Xiang-Li Li 4e7dbd327f
Update __init__.py
update version for nccl & mpi uint8
2021-09-25 20:53:40 +08:00
Xiang-Li Li 6eb1c77474
Update mpi_broadcast_op.cc
support uint8 in mpi
2021-09-25 20:49:29 +08:00
Xiang-Li Li beebddafd5
Update nccl_broadcast_op.cc
support uint8 in nccl
2021-09-25 20:48:18 +08:00
cxjyxx_me 764c8425af polish named_parameters 2021-09-22 22:27:13 -04:00
Dun Liang d85af13024 polish initialize order 2021-09-16 13:51:28 +08:00
Dun Liang 40cb853e21 update version 103 for windows 2021-09-15 17:48:59 +08:00
Dun Liang 500dfc6ee3 Merge branch 'master' of github.com:Jittor/jittor 2021-09-15 17:45:37 +08:00
Dun Liang c3938e14bf msvc support 2021-09-15 17:34:50 +08:00
Dun Liang 4fb462e1b9 polish sub module save load 2021-09-13 11:03:28 +08:00
Dun Liang 42dfaaed2e polish transform 2021-09-09 11:32:16 +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 36f09282fd optimize repeat 2021-09-08 10:38:20 +08:00
lzhengning 40ce5e85cd polish repeat: support higher dimension 2021-09-07 20:15:33 +08:00
Dun Liang 69325deb45 add emnist dataset and mac polish 2021-09-04 14:37:49 +08:00
Dun Liang 4e712f283f optimize cumsum with unroll 2021-09-02 17:22:46 +08:00
cxjyxx_me 546860e19e fix cumprod 2021-09-02 04:10:17 -04:00
cxjyxx_me 8a2e7a1881 fix 2021-09-02 04:09:29 -04:00
cxjyxx_me 010d306163 fix 2021-09-02 04:05:38 -04:00
cxjyxx_me 393b83e22c test cub cumsum 2021-09-02 04:02:40 -04:00
cxjyxx_me 5b830dcf66 cub cumsum 2021-09-02 04:02:31 -04:00
cxjyxx_me c53d473d2a fix cumsum 2021-09-02 04:02:22 -04:00
Jittor ef83872459
Update res2net.py 2021-09-02 13:24:01 +08:00
Jittor c90954a325
Update res2net.py 2021-09-02 12:25:32 +08:00
Dun Liang 655362efd7 better memory profiler 2021-09-02 11:20:25 +08:00
Jittor 8dd2520adc
Update res2net.py 2021-09-02 11:17:40 +08:00
Dun Liang aa749dd0e0 polish exit cleanup 2021-08-31 11:42:12 +08:00
cxjyxx_me 7349284663 Merge branch 'master' of github.com:Jittor/jittor 2021-08-26 22:28:06 -04:00
cxjyxx_me b9f97430c6 set different seed for different process&worker 2021-08-26 22:27:50 -04:00
Dun Liang 810781eae7 doc polish and normalize polish 2021-08-25 12:11:06 +08:00
cxjyxx_me 76090168d0 Merge branch 'master' of github.com:Jittor/jittor 2021-08-24 10:59:41 -04:00
cxjyxx_me 46955965d9 set_global_seed for worker 2021-08-24 10:58:09 -04:00
Dun Liang 1f16af33a2 polish mpi dataset shuffle list 2021-08-24 17:22:39 +08:00
Dun Liang 4e38190483 Merge branch 'master' of https://github.com/Jittor/jittor 2021-08-17 13:35:18 +08:00
Dun Liang 4fd355202c add naive windows support 2021-08-15 19:58:18 +08:00
Dun Liang 4217359f86 add fuseable transpose op 2021-08-12 20:56:03 +08:00
Dun Liang 1c2668cdb8 add float64 atomic 2021-08-11 19:16:15 +08:00
Dun Liang 692cbddb8e polish world rank and world size 2021-08-05 20:09:25 +08:00
Dun Liang 39165cccfb polish bmm 2021-08-04 19:36:02 +08:00
Dun Liang 86a3feeaab polish pool interface 2021-08-04 19:28:54 +08:00
li-xl 317e07907f polish transpose 2021-08-04 16:49:45 +08:00
li-xl aab4bda835 polish dataset 2021-08-04 16:49:45 +08:00
Dun Liang 1f0ea3b796 add mac mem_info support 2021-07-31 14:33:41 +08:00
lzhengning 71f862b898 add mac meminfo 2021-07-31 14:25:42 +08:00
Dun Liang 11c2812b5c prune pyc 2021-07-31 14:22:39 +08:00
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
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
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
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 95d85f29ef add backward hook 2021-07-04 17:35:09 +08:00
Dun Liang cf6a73c6a5 fix typo 2021-07-02 15:50:14 +08:00
Dun Liang 93940ebb31 add expand -1 interface 2021-07-02 15:42:11 +08:00
Dun Liang b4d6f9880d add lock for dl_open 2021-07-02 14:47:18 +08:00
lzhengning 3a5a78544f doc: set seed 2021-07-02 12:09:17 +08:00
Dun Liang 7a253e4a4a polish ddp with multiple workers 2021-07-01 21:16:45 +08:00