SkillAgentSearch skills...

AdderNetCuda

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

Install / Use

/learn @jdnie/AdderNetCuda
About this skill

Quality Score

0/100

Supported Platforms

Universal

README

运行方法

cd unoptimized

python setup.py install

cd ..

python train.py

说明

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

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

View on GitHub
GitHub Stars18
CategoryDevelopment
Updated1y ago
Forks0

Languages

Python

Security Score

75/100

Audited on Nov 22, 2024

No findings