You could manually push data to the CPU and let it perform some operations. Note however, that data transfers can be expensive and would most likely reduce the performance of your model. Also, the CPU is still executing the actual Python code and is launching the GPU workload, so it’s not completely idle,