Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Use 4 core laptop for smallmlp and update printed losses #103

Merged
merged 2 commits into from
Aug 29, 2022
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 28 additions & 8 deletions docs/src/examples/smallmlp.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,16 +54,36 @@ end
```
I get
```julia
julia> for _ in 1:3
@time SimpleChains.train_unbatched!(
G, p, mlpdloss, X, SimpleChains.ADAM(), 10_000
);
report(p)
end
5.258996 seconds (7.83 M allocations: 539.553 MiB, 4.18% gc time, 69.59% compilation time)
┌ Info: Loss:
│ train = 0.012996411f0
└ test = 0.021395735f0
0.488138 seconds
│ train = 1243.1248f0
└ test = 483.38852f0
1.638860 seconds
┌ Info: Loss:
│ train = 0.0027068993f0
└ test = 0.009439239f0
0.481226 seconds
│ train = 96.98259f0
└ test = 210.4579f0
1.654781 seconds
┌ Info: Loss:
│ train = 0.0016358295f0
└ test = 0.0074498975f0
│ train = 44.350838f0
└ test = 164.85913f0

julia> versioninfo()
Julia Version 1.9.0-DEV.1189
Commit 293031b4a5* (2022-08-26 20:24 UTC)
Platform Info:
OS: Linux (x86_64-redhat-linux)
CPU: 8 × 11th Gen Intel(R) Core(TM) i7-1165G7 @ 2.80GHz
WORD_SIZE: 64
LIBM: libopenlibm
LLVM: libLLVM-14.0.5 (ORCJIT, tigerlake)
Threads: 8 on 8 virtual cores
Environment:
JULIA_NUM_THREADS = 8
```