├── README.md ├── TerActivateFunc ├── grad_test_ter.py ├── install.sh └── src │ └── cuda │ ├── TerActivateFunc_cuda.cpp │ ├── TerActivateFunc_cuda_kernel.cu │ ├── __init__.py │ └── setup.py ├── evaluate.py ├── main.py ├── merge_from_bwx2_to_ternary.py ├── model_list ├── __init__.py ├── resnet.py ├── resnet18_bwx2.py └── resnet18_ter.py └── util.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/README.md -------------------------------------------------------------------------------- /TerActivateFunc/grad_test_ter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/TerActivateFunc/grad_test_ter.py -------------------------------------------------------------------------------- /TerActivateFunc/install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/TerActivateFunc/install.sh -------------------------------------------------------------------------------- /TerActivateFunc/src/cuda/TerActivateFunc_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/TerActivateFunc/src/cuda/TerActivateFunc_cuda.cpp -------------------------------------------------------------------------------- /TerActivateFunc/src/cuda/TerActivateFunc_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/TerActivateFunc/src/cuda/TerActivateFunc_cuda_kernel.cu -------------------------------------------------------------------------------- /TerActivateFunc/src/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /TerActivateFunc/src/cuda/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/TerActivateFunc/src/cuda/setup.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/evaluate.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/main.py -------------------------------------------------------------------------------- /merge_from_bwx2_to_ternary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/merge_from_bwx2_to_ternary.py -------------------------------------------------------------------------------- /model_list/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/model_list/__init__.py -------------------------------------------------------------------------------- /model_list/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/model_list/resnet.py -------------------------------------------------------------------------------- /model_list/resnet18_bwx2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/model_list/resnet18_bwx2.py -------------------------------------------------------------------------------- /model_list/resnet18_ter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/model_list/resnet18_ter.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WeixiangXu/STTN/HEAD/util.py --------------------------------------------------------------------------------