ToGPU
function defined in module
FluxTraining
ToGPU()
Callback that moves model and batch data to the GPU during training. Convenience for
ToDevice
(Flux.gpu).
There is
1
method for FluxTraining.ToGPU:
The following pages link back here:
Callback reference, How to use callbacks, Tips & tricks, Training an image classifier, imagenette_demo
FluxTraining.jl , callbacks/callbacks.jl , callbacks/sanitycheck.jl