Open ablaom opened 2 years ago
Now that we have https://github.com/FluxML/Flux.jl/pull/1849#event-5967026617, there are some things we could do:
rng
Short
Dropout
Short2
Now that we have https://github.com/FluxML/Flux.jl/pull/1849#event-5967026617, there are some things we could do:
rng
argument for theShort
builder should be passed to theDropout
layer at https://github.com/FluxML/MLJFlux.jl/blob/b5a9c419bb3af4aae18fdbd3fb69b03497b6ea9e/src/builders.jl#L61Short
instead of the test-local builderShort2
in tests and removeShort2
: https://github.com/FluxML/MLJFlux.jl/blob/b5a9c419bb3af4aae18fdbd3fb69b03497b6ea9e/test/regressor.jl#L6Dropout
layer with rng