-
Notifications
You must be signed in to change notification settings - Fork 50
Description
Hi,
I'm attempting to train a network consisting of a recurrent layer on top of a time delay conv net using nn.Recurrent. It runs fine, albeit slowly, on my Macbook, but when I try to run it on my much faster Ubuntu desktop I get a segmentation fault error with a core dump message. Where does torch dump the error message?
In case you want to try reproducing the error, the network looks like this:
memorySize = 10
windowSize = 5
nSamples = 2693
nFeatures = 100
nClasses = 7
TDCNN = nn.Sequential()
TDCNN:add(nn.LookupTable(nSamples, nFeatures))
TDCNN:add(nn.Padding(1, windowSize-1))
TDCNN:add(nn.Padding(1, -windowSize+1))
TDCNN:add(nn.TemporalConvolution(nFeatures, nFeatures, windowSize))
TDCNN:add(nn.Max(1))
RNN = nn.Recurrent(nn.Identity(), TDCNN, nn.Linear(nFeatures, nFeatures), nn.Tanh(), memorySize)
model = nn.Sequential()
model:add(RNN)
model:add(nn.Linear(nFeatures, nClasses))
model:add(nn.LogSoftMax())
loss = nn.ClassNLLCriterion()
Inputs to the network are torch.LongTensors of varying length.
Thanks,
Shawn