Basic utilities with few dependencies.
%load_ext autoreload
%autoreload 2
%matplotlib inline
At training time, we will typically want to put the model and the current mini batch on the GPU. When developing on a CPU, a GPU isn't available, so we define a variable that will automatically find the right device. This goes in utils rather than core to avoid circular imports with the callbacks module.
DEVICE