parent
4035745f6e
commit
043fedde3c
1
REQUIRE
1
REQUIRE
@ -9,6 +9,7 @@ GZip
|
||||
Colors
|
||||
ZipFile
|
||||
AbstractTrees
|
||||
Reexport
|
||||
|
||||
# AD
|
||||
ForwardDiff 0.5.0
|
||||
|
@ -4,7 +4,7 @@ module Flux
|
||||
|
||||
# Zero Flux Given
|
||||
|
||||
using Juno, Requires
|
||||
using Juno, Requires, Reexport
|
||||
using MacroTools: @forward
|
||||
|
||||
export Chain, Dense, RNN, LSTM, GRU, Conv2D,
|
||||
@ -12,9 +12,7 @@ export Chain, Dense, RNN, LSTM, GRU, Conv2D,
|
||||
SGD, ADAM, Momentum, Nesterov, AMSGrad,
|
||||
param, params, mapleaves
|
||||
|
||||
using NNlib
|
||||
export σ, sigmoid, logσ, logsigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax,
|
||||
conv2d, maxpool2d, avgpool2d
|
||||
@reexport using NNlib
|
||||
|
||||
include("tracker/Tracker.jl")
|
||||
using .Tracker
|
||||
|
Loading…
Reference in New Issue
Block a user