RuntimeError in cd_model_unc_cifar10_wine.ipynb example

[ascillitoe]

From @KevinRyu in #432:

In cd_model_unc_cifar10_wine.ipynb example, .cuda() should be added to reg model variable.
If this is not applied, I got below error.

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat1 in method wrapper_addmm)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
trainer(reg.cuda(), nn.MSELoss(), X_train_dl, device, torch.optim.Adam, learning_rate=0.001, epochs=30)