`ToGPU`

function defined in module FluxTraining


			ToGPU()

Callback that moves model and batch data to the GPU during training. Convenience for ToDevice(Flux.gpu).

Methods

There is 1 method for FluxTraining.ToGPU:

callbacks/callbacks.jl:119

Backlinks

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