zackchase / mxnet-the-straight-dope

An interactive book on deep learning. Much easy, so MXNet. Wow. [Straight Dope is growing up] ---> Much of this content has been incorporated into the new Dive into Deep Learning Book available at https://d2l.ai/.
https://d2l.ai/
Apache License 2.0
2.56k stars 724 forks source link

Correct context in Chapter 08 - Fine Tuning. #524

Closed vishaalkapoor closed 5 years ago

vishaalkapoor commented 5 years ago

This notebook will not run when there is a GPU because the network is copied to the GPUs, but the param data is pulled into the default context (CPU). Specify the context to fix the issue.