Optimisers.jl icon indicating copy to clipboard operation
Optimisers.jl copied to clipboard

Published 20 hours ago • verified publisher icon FluxML

Test with GPUArrays

Open mcabbott opened this issue 3 years ago • 0 comments

This adds tests which should detect problems with CuArrays, which can run on your laptop (and on CI) using a fake GPUArray. Similar to https://github.com/FluxML/OneHotArrays.jl/pull/9

~~WIP as it should still test gradients.~~

Xref https://github.com/FluxML/Optimisers.jl/issues/69 about destructure of a model containing both CPU and GPU arrays, not clear what should happen there ideally.

mcabbott avatar Apr 22 '22 21:04 mcabbott