how to train on cpu?
lonngxiang opened this issue · comments
As long as you do not specify in the command line arguments --gpus # then you will be training on CPU. I recommend you try using the train_finetune.py to see results without having to wait too long. It functions just the same as train.py except you do not need to specify the model_name in the command line.