Add transfer learning code with no GPU

FluxML / fluxml.github.io

Flux Website

https://fluxml.ai

MIT License

20 stars 45 forks source link

Add transfer learning code with no GPU #90

Open logankilpatrick opened 3 years ago

logankilpatrick commented 3 years ago

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

ToucheSir commented 3 years ago

gpu is a no-op when CUDA.jl is disabled, so this example should run as-is. That said, we could always add a code snippet or a couple sentences that talk about this.