From 9d5ffa2a97b32286af7b690884f86ffce7ad3cc2 Mon Sep 17 00:00:00 2001 From: Clement Farabet Date: Tue, 27 Sep 2011 14:14:27 -0400 Subject: Added KL Diverg criterioAdded KL Diverg criterionn --- init.lua | 1 + 1 file changed, 1 insertion(+) (limited to 'init.lua') diff --git a/init.lua b/init.lua index 4e694db..9815544 100644 --- a/init.lua +++ b/init.lua @@ -93,6 +93,7 @@ torch.include('nnx', 'SpatialConvolutionSparse.lua') torch.include('nnx', 'SuperCriterion.lua') torch.include('nnx', 'SparseCriterion.lua') torch.include('nnx', 'DistNLLCriterion.lua') +torch.include('nnx', 'KLDivCriterion.lua') torch.include('nnx', 'SpatialMSECriterion.lua') torch.include('nnx', 'SpatialClassNLLCriterion.lua') torch.include('nnx', 'SpatialSparseCriterion.lua') -- cgit v1.2.3