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:
FastAI.jl, How to log to TensorBoard, How to train a model, Image segmentation, Introduction, Keypoint regression, Quickstart, Saving and loading models for inference, Siamese image similarity, TimeSeries Classification, Variational autoencoders, tsregression
learner.jl , FluxTraining.jl , callbacks/callbacks.jl , callbacks/sanitycheck.jl