Unless the CPU code is un-runnable as is, we should give a CPU only option here: https://fluxml.ai/tutorials/2020/10/18/transfer-learning.html