Luxios22 / Dual_Norm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ROSE Lab Dual Norm

Training

python train.py config/dual_norm.yaml --DEVICE cuda:0

Testing

python test.py config/dual_norm.yaml

Testing Offical Weight

save to official weight as net_160.pth in the Dual_Norm_Official folder in checkpoint

python test.py config/dual_norm_official.yaml 18530

About

License:MIT License


Languages

Language:Python 100.0%