Training addernet accelerated by CUDA

Usage

cd adder_cuda
python setup.py install
cd ..
python main.py

Env

pytorch 1.10.0
CUDA 11.3

GitHub