From f7fdfbe3a98bab8b59208b84a8a22d7583b77320 Mon Sep 17 00:00:00 2001
From: Iblis Lin <iblis@hs.ntnu.edu.tw>
Date: Thu, 8 Feb 2018 12:55:16 +0000
Subject: [PATCH] fix `params`

---
 src/treelike.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/treelike.jl b/src/treelike.jl
index b4c9a77e..51391250 100644
--- a/src/treelike.jl
+++ b/src/treelike.jl
@@ -33,9 +33,11 @@ function prefor(f, x; seen = OSet())
   return
 end
 
+using Flux.Tracker: istracked
+
 function params(m)
   ps = []
-  prefor(p -> p isa TrackedArray && push!(ps, p), m)
+  prefor(p -> istracked(p) && push!(ps, p), m)
   return ps
 end