For the notebook to run on a GPU the data need to be sent to the GPU. To be...

For the notebook to run on a GPU the data need to be sent to the GPU. To be generally applicable there is now a fix implemented that makes the notebook work on on both CPUs and GPUs. In addition, there were two instances in the data were either the label or the class were not well defined making the final test fail. These are now skipped.