From ba152e884d7c521a663a819131d029938a7023aa Mon Sep 17 00:00:00 2001 From: Clement Farabet Date: Thu, 5 Jul 2012 19:44:29 -0400 Subject: Added experimental Diagonal Hessian code. Deactivated by default: nn.hessian.activate() appends all the necessary functions to use second order information for training. --- init.lua | 1 + 1 file changed, 1 insertion(+) (limited to 'init.lua') diff --git a/init.lua b/init.lua index c1c7b79..e5deccf 100644 --- a/init.lua +++ b/init.lua @@ -92,4 +92,5 @@ torch.include('nn', 'MultiLabelMarginCriterion.lua') torch.include('nn', 'StochasticGradient.lua') torch.include('nn', 'Jacobian.lua') +torch.include('nn', 'hessian.lua') torch.include('nn', 'test.lua') -- cgit v1.2.3