Error regarding multiple devices 

Im having this Error:
```
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
```
It can be fixed by adding:
```
.optOption("mapLocation", "true") 
.optDevice(Device.gpu())
```
to the criteria.
regards,
Felix