Merge branch 'master' of github.com:pluskid/Mocha.jl

pluskid · Feb 24, 2015 · e14801b · e14801b
2 parents 1c73471 + 299f9e6
commit e14801b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/net.jl b/src/net.jl
@@ -61,12 +61,16 @@ end
 function freeze!(net::Net) end
 function freeze!(net::Net, idx::Int...)
   for i in idx
-    freeze!(get_layer_state(net, i))
+    layer_state = get_layer_state(net, i)
+    @info("Freezing layer $(layer_state.layer.name) in network $(net.name)...")
+    freeze!(layer_state)
   end
 end
 function freeze!(net::Net, names::String...)
   for name in names
-    freeze!(get_layer_state(net, name))
+    layer_state = get_layer_state(net, name)
+    @info("Freezing layer $(layer_state.layer.name) in network $(net.name)...")
+    freeze!(layer_state)
   end
 end