Skip to content

Instantly share code, notes, and snippets.

@madisonmay
Created August 22, 2014 03:46
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save madisonmay/a016d6c305fb4e018198 to your computer and use it in GitHub Desktop.
Save madisonmay/a016d6c305fb4e018198 to your computer and use it in GitHub Desktop.
Reproduce behavior described in pylearn2 issue #1083: https://github.com/lisa-lab/pylearn2/issues/1083
!obj:pylearn2.train.Train {
dataset: &train !obj:pylearn2.datasets.mnist.MNIST {
which_set: 'train',
start: 0,
stop: 50000
},
model: !obj:pylearn2.models.mlp.MLP {
layers: [
!obj:pylearn2.models.mlp.Softmax {
layer_name: 'y',
n_classes: 10,
irange: 0.0,
init_bias_target_marginals: *train
}],
nvis: 784
},
algorithm: !obj:pylearn2.training_algorithms.sgd.SGD {
batch_size: 100,
learning_rate: 0.01,
monitoring_dataset: {
'train': *train,
'valid': !obj:pylearn2.datasets.mnist.MNIST {
which_set: 'train',
start: 50000,
stop: 51000
}
},
termination_criterion: !obj:pylearn2.termination_criteria.EpochCounter {
max_epochs: 100
}
}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment