Merge pull request #89 from sisl/patch/flux

remove mention of Tracker to keep up with Flux
sisl · Feb 28, 2020 · 4135f34 · 4135f34
2 parents 606a5d6 + 7e79138
commit 4135f34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/flux.jl b/src/utils/flux.jl
@@ -9,7 +9,7 @@ activation(::typeof(relu)) = ReLU()
 
 layer(x) = error("Can't use $x as a layer.")
 
-layer(d::Dense) = Layer(Tracker.data(d.W), Tracker.data(d.b), activation(d.σ))
+layer(d::Dense) = Layer(data(d.W), data(d.b), activation(d.σ))
 
 network(c::Chain) = Network([layer.(c.layers)...])