From 5edaa7d5649a2e124496159f7e8a14edab4f7484 Mon Sep 17 00:00:00 2001 From: Determinant Date: Sat, 30 May 2015 20:52:05 +0800 Subject: rename param file to chunk file (intend to generalize it) --- layer/affine.lua | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'layer/affine.lua') diff --git a/layer/affine.lua b/layer/affine.lua index 97703a8..d88813e 100644 --- a/layer/affine.lua +++ b/layer/affine.lua @@ -3,7 +3,8 @@ local BiasParam = nerv.class('nerv.BiasParam', 'nerv.LinearTransParam') local AffineLayer = nerv.class('nerv.AffineLayer', 'nerv.Layer') function LinearTransParam:read(pcdata) - self.trans = nerv.CuMatrixFloat.new_from_host(nerv.MMatrixFloat.load(pcdata)) + self.trans = self.gconf.mat_type.new_from_host( + nerv.MMatrixFloat.load(pcdata)) end function LinearTransParam:write(pfhandle) -- cgit v1.2.3