FluxML / FluxTraining.jl

A flexible neural net training library inspired by fast.ai
https://fluxml.ai/FluxTraining.jl
MIT License
118 stars 25 forks source link

Add training loop profiler #89

Open lorenzoh opened 2 years ago

lorenzoh commented 2 years ago

This implements the training loop profiler suggested in #86. See also for a discussion about benchmarking: https://github.com/FluxML/ML-Coordination-Tracker/discussions/40.

The profiling part is implemented, but functionality to make it easier to perform analyses is still TODO.