-
Notifications
You must be signed in to change notification settings - Fork 16
Open
Description
Now that we have FluxML/Flux.jl#1849 (comment), there are some things we could do:
- The
rngargument for theShortbuilder should be passed to theDropoutlayer atLine 61 in b5a9c41
Flux.Dropout(builder.dropout), - Use
Shortinstead of the test-local builderShort2in tests and removeShort2:Line 6 in b5a9c41
# TODO: replace Short2 -> Short when - Update this section of the README.md: https://github.com/FluxML/MLJFlux.jl#random-number-generators-and-reproducibility. We should add sentence explaining limitations of Dropout RNG's for GPU training (and copy @darsnack in the PR to check we have this correct)
- In the README.md example at https://github.com/FluxML/MLJFlux.jl#random-number-generators-and-reproducibility, include a
Dropoutlayer with rng
darsnack
Metadata
Metadata
Assignees
Labels
No labels