inplace_abn:就地激活的BatchNorm，用于DNN的内存优化训练
需积分: 45 30 浏览量
2021-04-06
11:24:05
上传
评论 1
收藏 149KB ZIP 举报
inplace_abn-master.zip （60个子文件） 
inplace_abn-master 
MANIFEST.in 33B
inplace_abn.png 85KB src inplace_abn_kernels.cuh 11KB inplace_abn.cpp 7KB inplace_abn_cpu.cpp 6KB inplace_abn_cuda.cu 14KB utils.cpp 500B scripts train_imagenet.py 11KB experiments resnext152_ipabn_lr_256.json 459B wider_resnet38_ipabn_lr_256.json 514B resnet101_ipabn-sync_lr_512.json 433B resnext101_ipabn-sync_lr_256.json 456B resnet34_ipabn-sync_lr_512.json 432B densenet264_ipabn_lr_256.json 460B resnext101_stdbn_lr_256.json 460B resnet50_ipabn-sync_lr_512.json 432B resnext101_ipabn_lr_512.json 459B models util.py 128B wider_resnet.py 7KB resnext.py 5KB __init__.py 97B densenet.py 4KB resnet.py 4KB test_imagenet.py 3KB modules deeplab.py 3KB misc.py 550B residual.py 7KB __init__.py 166B dense.py 1KB
requirements.txt 43B imagenet config.py 2KB utils.py 11KB __init__.py 0B transforms.py 2KB test_vistas_single_gpu.py 11KB dataset transform.py 763B dataset.py 1KB __init__.py 0B sampler.py 2KB test_vistas.py 2KB requirements.txt 100B CONTRIBUTING.md 697B inplace_abn functions.py 11KB __init__.py 183B group.py 2KB abn.py 11KB _backend.pyi 1KB LICENSE 2KB setup.cfg 34B setup.py 2KB README.md 12KB licenses.csv 192B
equation.svg 5KB CODE_OF_CONDUCT.md 3KB .gitignore 982B include dispatch.h 3KB cuda_utils.cuh 4KB inplace_abn.h 4KB checks.h 2KB utils.h 1KB资源评论
邱笑晨
- 粉丝: 39
- 资源: 4553
