场景模型>> AdderNetCuda>> 返回
项目作者： jdnie

项目描述 ：
CUDA实现'huawei-noah/AdderNet'的forward和backward
高级语言： Python
项目地址: git://github.com/jdnie/AdderNetCuda.git
创建时间： 2020-04-16T03:13:53Z
项目社区：https://github.com/jdnie/AdderNetCuda
开源协议：BSD 3-Clause "New" or "Revised" License
下载


运行方法

cd unoptimized

python setup.py install

cd ..

python train.py

说明

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

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