Skip to content

Commit

Permalink
Make example code fit on page
Browse files Browse the repository at this point in the history
  • Loading branch information
Kaixhin committed Dec 22, 2015
1 parent b6cfeb9 commit 5f896b1
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,10 @@ cnn:add(cudnn.SpatialConvolution(nChannels, nFilters, 2, 2):init('weight', nnini
:init('weight', nninit.addNormal, 0, 0.01)
:init(getBias, nninit.constant, 0))
cnn:add(nn.View(nFilters*15*15))
cnn:add(nn.Linear(nFilters*15*15, nFilters):init('weight', nninit.kaiming, {dist = 'uniform', gain = {'lrelu', leakiness = 0.3}}))
cnn:add(nn.Linear(nFilters*15*15, nFilters):init('weight', nninit.kaiming, {
dist = 'uniform',
gain = {'lrelu', leakiness = 0.3}
}))
cnn:add(nn.RReLU(1/3, 1/3))
cnn:add(nn.Linear(nFilters, 6):init('weight', nninit.orthogonal, {gain = 'relu'}))
cnn:add(cudnn.ReLU())
Expand Down

0 comments on commit 5f896b1

Please # to comment.