Skip to content
Branch: master
Find file History
ottonemo and BenjaminBossan Move criterion to compute device automatically (#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
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
callbacks [MRG] Warn user when using possibly wrong checkpoint monitor (#403) Dec 11, 2018
tests
__init__.py WIP: Update to 0.4.1 (#298) Aug 1, 2018
classifier.py
cli.py Tests pass even when fire and numpydoc are not installed. (#411) Dec 13, 2018
dataset.py Better error msg when dataset incompatible (#427) Feb 7, 2019
exceptions.py Correct backtick usage in docstrings. Oct 13, 2017
helper.py Feature: quick CLIs (#390) Dec 13, 2018
history.py [MRG] Adds optimizer and history to save/load_params (#360) Oct 22, 2018
net.py Move criterion to compute device automatically (#455) Apr 15, 2019
regressor.py
setter.py
toy.py Fix minor error in toy documentation (#452) Apr 5, 2019
utils.py Add support for scipy sparse CSR matrices as input data. (#406) Dec 11, 2018
You can’t perform that action at this time.