AdderNetCuda
CUDA实现'huawei-noah/AdderNet'的forward和backward
Install / Use
/learn @jdnie/AdderNetCudaREADME
运行方法
cd unoptimized
python setup.py install
cd ..
python train.py
说明
cuda代码在torch=1.2.0,torchvision=0.4.0环境下调试ok。
AdderNet不太容易收敛,可能实现哪里有bug吧。
CUDA实现'huawei-noah/AdderNet'的forward和backward
/learn @jdnie/AdderNetCudacd unoptimized
python setup.py install
cd ..
python train.py
cuda代码在torch=1.2.0,torchvision=0.4.0环境下调试ok。
AdderNet不太容易收敛,可能实现哪里有bug吧。