From a54332ce81129e81fbb1d041ec41aa5955868c5e Mon Sep 17 00:00:00 2001 From: Determinant Date: Fri, 11 Mar 2016 17:33:35 +0800 Subject: adapt asr_trainer.lua to new architecture --- nerv/layer/sigmoid.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'nerv/layer/sigmoid.lua') diff --git a/nerv/layer/sigmoid.lua b/nerv/layer/sigmoid.lua index a9f9749..5974ffc 100644 --- a/nerv/layer/sigmoid.lua +++ b/nerv/layer/sigmoid.lua @@ -3,6 +3,9 @@ local SigmoidLayer = nerv.class("nerv.SigmoidLayer", "nerv.Layer") function SigmoidLayer:__init(id, global_conf, layer_conf) nerv.Layer.__init(self, id, global_conf, layer_conf) self:check_dim_len(1, 1) + if self.dim_in[1] ~= self.dim_out[1] then + nerv.error("mismatching dimensions of input and output") + end end function SigmoidLayer:bind_params() @@ -10,9 +13,6 @@ function SigmoidLayer:bind_params() end function SigmoidLayer:init() - if self.dim_in[1] ~= self.dim_out[1] then - nerv.error("mismatching dimensions of input and output") - end end function SigmoidLayer:batch_resize(batch_size) -- cgit v1.2.3-70-g09d2