Skip to content

Commit a92606b

Browse files
fix adamw
1 parent 1930966 commit a92606b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/optimise/optimisers.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ opt = ADAMW(0.001, (0.89, 0.995), 0.1)
500500
```
501501
"""
502502
ADAMW= 0.001, β = (0.9, 0.999), decay = 0) =
503-
Optimiser(ADAM(1, β), WeightDecay(decay), Descent))
503+
Optimiser(ADAM(η, β), WeightDecay(decay))
504504

505505
"""
506506
AdaBelief(η = 0.001, β::Tuple = (0.9, 0.999), ϵ = $EPS)

0 commit comments

Comments
 (0)