Train convolutional FISTA

rgoroshin / LISTA

LISTA/dictionary learning

12 stars 10 forks source link

Train convolutional FISTA #3

Open jakezhaojb opened 9 years ago

jakezhaojb commented 9 years ago

line 33: ConvDec = nn.NormSpatialConvolutionFFT(outplane, inplane, k, k, stride, stride) line 44: ConvEnc = nn.SpatialConvolutionFFT(inplane, outplane, k, k, stride, stride)

Shouldn't we do valid conv followed by full conv around these two lines?