| .. |
|
dataset
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
extern
|
support package installed cuda
|
2021-05-27 19:36:17 +08:00 |
|
models
|
add res2net pretrain
|
2021-04-03 15:30:47 +08:00 |
|
notebook
|
move files
|
2021-05-16 16:08:31 +08:00 |
|
script
|
move files
|
2021-05-16 16:08:31 +08:00 |
|
src
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
test
|
add clip_grad_norm
|
2021-05-13 22:03:20 +08:00 |
|
transform
|
fix transforms.xxxresize int_size (#200)
|
2021-04-20 17:03:19 +08:00 |
|
utils
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
__init__.py
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
attention.py
|
update 2020 -> 2021
|
2021-01-07 16:25:00 +08:00 |
|
compile_extern.py
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
compiler.py
|
auto cuda downloader
|
2021-05-28 15:46:05 +08:00 |
|
contrib.py
|
polish getitem
|
2021-03-12 15:40:29 +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
|
polish distributions
|
2021-05-12 21:32:39 +08:00 |
|
init.py
|
fix init no_grad
|
2021-04-17 20:44:35 +08:00 |
|
init_cupy.py
|
failback to mpi if nccl not available
|
2021-05-14 13:45:38 +08:00 |
|
linalg.py
|
add unfold,fold,bicubic,qr,eye (#180)
|
2021-03-02 16:38:28 +08:00 |
|
lr_scheduler.py
|
update lr according to param_group["lr"] (#210)
|
2021-05-12 21:51:38 +08:00 |
|
misc.py
|
add distribution for rl
|
2021-04-29 23:04:02 +08:00 |
|
nn.py
|
add UpsamplingBilinear2d, AdaptiveMaxPool2d
|
2021-05-11 16:29:59 +08:00 |
|
optim.py
|
add clip_grad_norm
|
2021-05-13 22:03:20 +08:00 |
|
pool.py
|
add UpsamplingBilinear2d, AdaptiveMaxPool2d
|
2021-05-11 16:29:59 +08:00 |
|
pyjt_compiler.py
|
doc: optimize to reduce duplicated docs
|
2021-03-16 15:42:53 +08:00 |
|
sparse.py
|
update 2020 -> 2021
|
2021-01-07 16:25:00 +08:00 |
|
version
|
add centos support
|
2021-05-13 21:25:13 +08:00 |
Dun Liang
07eb4a7a0e