ottonemo and BenjaminBossan #455
Move criterion to compute device automatically Previously the user had to make sure that criterion parameters such as class weights are on the correct computing device. Now the criterion is moved to the compute device using `.to` which moves the parameters as well. Non nn.Module classes still work as criterion.
Latest commit 81ae225
Apr 15, 2019
|Type||Name||Latest commit message||Commit time|
|Failed to load latest commit information.|