How do I use the CPU for training?

[lvsejunzhuang]

Because my desktop computer does not have a GPU, I would like to try using a CPU for training

[mahdizynali]

i did not test it before but in "tools/test.py" there are some lines in order to check if gpu is available or not : if cfg.device.gpu_ids == -1: logger.info("Using CPU training") accelerator, devices = "cpu", None

and in the train.py it would check that argument :

if cfg.device.gpu_ids == -1: logger.info("Using CPU training") accelerator, devices, strategy, precision = ( "cpu", None, None, cfg.device.precision, )

so guess it will automatically switch on cpu mode if there is no any gpu however it's not recommended .

[ForeverPs]

set gpu_ids: "-1" in config file.