GPU CPU changing backend manually

Is there an easy way to switch from gpu to CPU after running one round of training ?

I need to build features for a new model using the original trained models but it takes really long on the GPU

Is there an easy way to switch between backends programmatically ?