From 1ee64da4abbafd92c6bea5121e9a11e55ef1cbcb Mon Sep 17 00:00:00 2001 From: koray kavukcuoglu Date: Thu, 3 Jan 2013 14:11:51 +0000 Subject: New NN classes extra/nn/L1Cost.lua : L1 penalty extra/nn/SpatialFullConvolution.lua : full convolution extra/nn/SpatialFullConvolutionMap.lua : full convolution with connection table extra/nn/TanhShrink.lua : shrinkage with x-tanh(x) extra/nn/WeightedMSECriterion.lua : mean squared error with weighting mask on the target Add new nn classes that are used commonly for unsupervised training of convolutional auto encoders --- init.lua | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'init.lua') diff --git a/init.lua b/init.lua index d121be1..587412f 100644 --- a/init.lua +++ b/init.lua @@ -45,6 +45,7 @@ include('SoftMin.lua') include('SoftPlus.lua') include('SoftSign.lua') include('Tanh.lua') +include('TanhShrink.lua') include('Abs.lua') include('Power.lua') include('Square.lua') @@ -55,6 +56,8 @@ include('Threshold.lua') include('LookupTable.lua') include('SpatialConvolution.lua') +include('SpatialFullConvolution.lua') +include('SpatialFullConvolutionMap.lua') include('SpatialConvolutionMM.lua') include('SpatialConvolutionMap.lua') include('SpatialSubSampling.lua') @@ -90,6 +93,8 @@ include('CosineEmbeddingCriterion.lua') include('MarginRankingCriterion.lua') include('MultiMarginCriterion.lua') include('MultiLabelMarginCriterion.lua') +include('L1Cost.lua') +include('WeightedMSECriterion.lua') include('StochasticGradient.lua') -- cgit v1.2.3