how to run this code with cpu

[Chenhait]

hi, thank you for your wonderful job!!!
I try to run this code on cpu.

I changed
profiling: [cpu]
in s_resnet50_train_val.yml

and run
python train.py app:apps/s_resnet50_train_val.yml

however, it still error
RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

is there other place should change?

thank you again!!
best