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

Published 20 hours ago • verified publisher icon JuliaML

Gpu of shuffleobs?

Open drozzy opened this issue 4 years ago • 1 comments

It seems if I have a shuffleobs and then wrap it in Flux.DataLoader, doing the |> gpu no longer moves the data to gpu :-(

drozzy avatar Dec 06 '20 02:12 drozzy

This seems to be a Flux-side issue. CuArray does doing the work.

johnnychen94 avatar Dec 06 '20 07:12 johnnychen94