| .. |
|
dataset
|
add jittor_home
|
2022-03-22 20:28:43 +08:00 |
|
demo
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
extern
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
loss3d
|
loss3d: chamfer & emd
|
2021-06-30 15:22:43 +08:00 |
|
models
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
notebook
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
other
|
polish fp16
|
2022-03-18 12:56:24 +08:00 |
|
script
|
fix windows encoding bugs
|
2022-03-09 18:23:42 +08:00 |
|
src
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
test
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
transform
|
fix: to_tensor applied on a jt.Var
|
2021-11-09 23:00:57 +08:00 |
|
utils
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
__init__.py
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
__init__.pyi
|
Merge branch 'master' into fft
|
2022-03-26 23:30:35 -04:00 |
|
attention.py
|
update 2020 -> 2021
|
2021-01-07 16:25:00 +08:00 |
|
compile_extern.py
|
polish torch issue
|
2022-04-05 16:55:01 +08:00 |
|
compiler.py
|
add amp alias
|
2022-03-26 15:51:47 +08:00 |
|
contrib.py
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
depthwise_conv.py
|
add stride padding(h,w) support and ctrl-c quick exit
|
2021-02-20 16:34:24 +08:00 |
|
distributions.py
|
floor keep type and better mem_info
|
2021-07-30 20:40:21 +08:00 |
|
init.py
|
Update trunc_normal_ with var = var.add(XX) and var = var.clamp(XX)
|
2022-03-22 20:48:41 +08:00 |
|
init_cupy.py
|
failback to mpi if nccl not available
|
2021-05-14 13:45:38 +08:00 |
|
linalg.py
|
polish einsum(phy)
|
2022-04-05 17:19:14 +08:00 |
|
lr_scheduler.py
|
add exp lr and remove warning as error
|
2021-07-26 21:21:20 +08:00 |
|
misc.py
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
nn.py
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
optim.py
|
v 1.3.3 memory optimization
|
2022-04-22 15:04:06 +08:00 |
|
pool.py
|
add fp16 support
|
2022-03-15 17:45:39 +08:00 |
|
pyjt_compiler.py
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
sparse.py
|
optimize concat and split
|
2022-04-04 22:53:20 +08:00 |
|
version
|
avoid import error when no torch installed
|
2021-06-18 15:56:13 +08:00 |
Dun Liang
0666456a2f