Skip to content

Commit

Permalink
Merge branch 'master' into data-parallelism
Browse files Browse the repository at this point in the history
  • Loading branch information
pluskid committed Dec 4, 2014
2 parents 0060b02 + 0c37e70 commit dd427cf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/mnist/mnist.jl
Expand Up @@ -6,7 +6,7 @@ ENV["MOCHA_USE_CUDA"] = "true"
using Mocha
srand(12345678)

data_layer = HDF5DataLayer(name="train-data", source="data/train.txt", batch_size=64)
data_layer = HDF5DataLayer(name="train-data", source="data/train.txt", batch_size=64, shuffle=true)
conv_layer = ConvolutionLayer(name="conv1", n_filter=20, kernel=(5,5), bottoms=[:data], tops=[:conv])
pool_layer = PoolingLayer(name="pool1", kernel=(2,2), stride=(2,2), bottoms=[:conv], tops=[:pool])
conv2_layer = ConvolutionLayer(name="conv2", n_filter=50, kernel=(5,5), bottoms=[:pool], tops=[:conv2])
Expand Down

0 comments on commit dd427cf

Please sign in to comment.