GitHub - jdnie/AdderNetCuda: CUDA实现'huawei-noah/AdderNet'的forward和backward

cd unoptimized

python setup.py install

cd ..

python train.py

cuda代码在torch=1.2.0，torchvision=0.4.0环境下调试ok。

AdderNet不太容易收敛，可能实现哪里有bug吧。

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
figures		figures
unoptimized		unoptimized
AdderNet.md		AdderNet.md
License.txt		License.txt
README.md		README.md
THIRD PARTY OPEN SOURCE SOFTWARE NOTICE.txt		THIRD PARTY OPEN SOURCE SOFTWARE NOTICE.txt
adder.py		adder.py
resnet20.py		resnet20.py
resnet50.py		resnet50.py
test.py		test.py
train.py		train.py
utils.py		utils.py

Provide feedback