As the criterion is not part of the model but instead a hyperparameter
due to the use of configuration files, it is not moved to the GPU if
selected as a device. We therefore manually move the criterion to the proper
device, which is bad practice when using lightning but works.

The obects.inv file from lightning's documentation is missing a lot of
methods, raising errors when building with sphinx. Problematic methods
have been added to the extras.txt and encoded into extras.inv