diff options
Diffstat (limited to 'nerv/layer/graph.lua')
| -rw-r--r-- | nerv/layer/graph.lua | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/nerv/layer/graph.lua b/nerv/layer/graph.lua index 1406eff..5f42fca 100644 --- a/nerv/layer/graph.lua +++ b/nerv/layer/graph.lua @@ -1,10 +1,7 @@ local GraphLayer = nerv.class('nerv.GraphLayer', 'nerv.Layer') function GraphLayer:__init(id, global_conf, layer_conf) - self.id = id - self.dim_in = layer_conf.dim_in - self.dim_out = layer_conf.dim_out - self.gconf = global_conf + nerv.Layer.__init(self, id, global_conf, layer_conf) self:graph_init(layer_conf.layer_repo, layer_conf.connections) end @@ -155,5 +152,5 @@ function GraphLayer:get_params() table.insert(param_repos, ref.layer:get_params()) end end - return nerv.ParamRepo.merge(param_repos) + return nerv.ParamRepo.merge(param_repos, self.loc_type) end |