deal with empty Chain

2019-09-10 10:46:56 -07:00 · 2019-09-10 10:46:56 -07:00 · 99679f7e16
commit 99679f7e16
parent d0202a2945
1 changed files with 5 additions and 3 deletions
--- a/src/layers/basic.jl
+++ b/src/layers/basic.jl
@ -51,9 +51,11 @@ Calculate the forward results of each layers in Chain `c` with `input` as model
 """
 function activations(c::Chain, input)
  res = Zygote.Buffer([], length(c))
-  res[1] = c[1](input)
-  for (i,l) in enumerate(c[2:end])
-    res[i+1] = l(res[i])
+  if length(c) > 0
+    res[1] = c[1](input)
+    for (i,l) in enumerate(c[2:end])
+      res[i+1] = l(res[i])
+    end
  end
  return copy(res)
 end