Skip to content
Snippets Groups Projects
Commit 507a6207 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Move criterion to selected device 

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.
parent e26fe458
No related branches found
No related tags found
1 merge request!4Moved code to lightning
Checking pipeline status
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment