search

lufficc / SSD

High quality, fast, modular reference implementation of SSD in PyTorch
MIT License
1.51k stars 385 forks source link

Hello,My PC is not equipped with CUDA, can I train my datasets by CPU ??........ #198

Open 605137155 opened 3 years ago

605137155 commented 3 years ago

My PC is not equipped with CUDA, can I train my datasets only by CPU ??........

605137155 commented 3 years ago

I know it! I only need to take place the "cuda" as"cpu " in ssd/condig/defaults.py where _c.MODEL.DEICE= "cuda" !!!hhh

FathiMahdi commented 3 years ago

Thank you was very helpful