Skip to content

Commit 6728d55

Browse files
committed
mention TerminalLoggers
1 parent f6baebb commit 6728d55

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

docs/src/training/train_api.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,10 @@ Flux.Train.setup
55
Flux.Optimise.train!(loss, model, data, opt; cb)
66
```
77

8+
`train!` uses [`@progress`](https://github.com/JuliaLogging/ProgressLogging.jl) which should show a progress bar in VSCode.
9+
To see one in a terminal, you will need to install [TerminalLoggers.jl](https://github.com/JuliaLogging/TerminalLoggers.jl)
10+
and follow its setup instructions.
11+
812
The new version of Flux's training code was written as an independent package, called Optimisers.jl.
913
However, at present all Flux models contain parameter arrays (such as `Array`s and `CuArray`s)
1014
which can be updated in-place. Thus objects returned by `update!` can be ignored.

0 commit comments

Comments
 (0)