diff options
Diffstat (limited to 'Optimization.lua')
-rw-r--r-- | Optimization.lua | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/Optimization.lua b/Optimization.lua index ed230e5..f18c635 100644 --- a/Optimization.lua +++ b/Optimization.lua @@ -3,10 +3,11 @@ local Optimization = torch.class('nn.Optimization') function Optimization:__init() end -function Optimization:forward(parameters, gradParameters) +function Optimization:forward(inputs, targets) self:flatten(parameters, gradParameters) - -- do your thing + self.output = 0 self:unflatten(parameters, gradParameters) + return self.output end function Optimization:flatten(parameters, gradParameters) |