Skip to content

Instantly share code, notes, and snippets.

@JoostvDoorn
Created September 20, 2016 13:44
Show Gist options
  • Save JoostvDoorn/dfa5f942ab5d01505d1b1fa4a88aa7f5 to your computer and use it in GitHub Desktop.
Save JoostvDoorn/dfa5f942ab5d01505d1b1fa4a88aa7f5 to your computer and use it in GitHub Desktop.
-- local parallel = nn.ConcatTable():add(nn.SelectTable(2)):add(nn.SelectTable(3))
-- self.baselineModel:add(parallel)
-- local lookup = nn.LookupTableMaskZero(vocabSize, 256)
-- local lookups = nn.ParallelTable():add(lookup):add(lookup:clone('weight','bias','gradWeight','gradBias'))
-- self.baselineModel:add(lookups)
-- local rnn1 = nn.Sequential()
-- rnn1.lstmLayers = {cudnn.GRU(256, 128, 1)}
-- rnn1:add(rnn1.lstmLayers[1])
-- local rnn2 = nn.Sequential()
-- rnn2.lstmLayers = {cudnn.GRU(256, 128, 1)}
-- rnn2:add(rnn2.lstmLayers[1])
-- self.baselineModel:add(cudnn.Connect(rnn1, rnn2))
-- self.baselineModel:add(nn.SelectTable(2))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment