search

beacon-biosignals / LighthouseFlux.jl

An adapter package that implements Lighthouse's framework interface for Flux
MIT License
1 stars 1 forks source link

force loss/prediction passed to Lighthouse to be moved to CPU #7

Closed jrevels closed 4 years ago

jrevels commented 4 years ago

AFAICT we might as well force this automatically if Lighthouse metrics aren't GPU-amenable anyway. Would be good for folks to test this out before we merge it, just in case.

Not really feasible to unit test this until we have GPU-enabled CI.