jdnie / AdderNetCuda

CUDA实现'huawei-noah/AdderNet'的forward和backward

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

运行方法

cd unoptimized

python setup.py install

cd ..

python train.py

说明

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

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

About

CUDA实现'huawei-noah/AdderNet'的forward和backward

License:BSD 3-Clause "New" or "Revised" License


Languages

Language:Python 64.3%Language:Cuda 31.7%Language:C++ 4.0%