search

CaptchaAgent / hcaptcha-model-factory

🏗 hCaptcha image label binary model factory (PyTorch Training, Cluster-based Auto Label Tools, Export ONNX model, ONNX model inference)
GNU General Public License v3.0
73 stars 18 forks source link

problem #2

Closed ninjadoc107 closed 2 years ago

ninjadoc107 commented 2 years ago

Torch not compiled with CUDA enabled

beiyuouo commented 2 years ago

I have updated the code. You can select your device in config now.