Closed khornlund closed 5 years ago
Very nice work! Thanks for doing this. I also couldn't figure out how to skip CUDA tests, and just got used to the idea of them failing when there's no GPU available, which isn't perfect but doesn't seem too bad.
I agree that more detailed examples are needed. I can make a "hello world" regression demo script, though I've never written a pytorch optimization loop so it may be a bit weird. That should resolve some of the confusion in the other thread about how to include parameters.
Re: #6
Summary of changes:
setup.py
file to enablesetuptools
installpartition_spline.npz
toresources
submodule, and modified code to usepkg_resources
to access it.tests/resources/
_get_device
toadaptive_test.py
to handle selection of torch device (I was getting errors trying to run the tests without the number on the end of 'cuda:{num}').Notes:
nosetest
way to skip the CUDA tests if no GPU is available. Is this possible?Cheers, Karl