FluxTraining.jl

Search Ctrl+K

Pages
Overview
Getting started
Features
Tutorials
Train an image classifier
Write a custom callback
How to
Use callbacks
Tips & Tricks
Reference
Module
List of callbacks
Training loop API
Backround
Hyperparameter scheduling
Custom training loops

Opened panes
Sliding panes disabled

Versions

links

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
FluxTraining module
CancelEpochException struct
CancelFittingException struct
CancelStepException struct
Checkpointer struct
ConditionalCallback struct
CustomCallback struct
EarlyStopping struct
Learner struct
LogHistograms struct
LogHyperParams struct
LogMetrics struct
LogTraces 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
Traces struct

Source files

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