FluxTraining

FluxTraining is a module

Symbols

Name Kind
AbstractCallback abstract type
LearningRate abstract type
GarbageCollect function
SmoothLoss function
ToGPU function
accuracy function
epoch! function
fit! function
loadmodel function
onecycle function
savemodel function
step! function
throttle function
CancelEpochException struct
CancelFittingException struct
CancelStepException struct
Checkpointer struct
ConditionalCallback struct
CustomCallback struct
EarlyStopping struct
Learner struct
LogHistograms struct
LogHyperParams struct
LogMetrics struct
LogVisualization struct
Loss struct
Metric struct
Metrics struct
MetricsPrinter struct
ProgressPrinter struct
Recorder struct
SanityCheck struct
Scheduler struct
StopOnNaNLoss struct
TensorBoardBackend struct
ToDevice struct

Source files

FluxTraining.jl
functional/metrics.jl
callbacks/protect.jl
callbacks/phases.jl
callbacks/events.jl
callbacks/callback.jl
callbacks/graph.jl
callbacks/execution.jl
callbacks/logging/Loggables.jl
callbacks/logging/logger.jl
callbacks/logging/tensorboard.jl
callbacks/logging/checkpointer.jl
callbacks/conditional.jl
callbacks/callbacks.jl
callbacks/earlystopping.jl
callbacks/custom.jl
callbacks/metrics.jl
callbacks/recorder.jl
callbacks/sanitycheck.jl
callbacks/hyperparameters.jl
callbacks/scheduler.jl
learner.jl
callbackutils.jl
training.jl
testutils.jl