Skip to content

Commit 91bee6f

Browse files
committed
Merge pull request lisa-lab#38 from renaud/patch-1
Update DBN.py
2 parents b88e1d3 + 00e98ff commit 91bee6f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Diff for: code/DBN.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,8 @@ def __init__(self, numpy_rng, theano_rng=None, n_ins=784,
4343
:type n_ins: int
4444
:param n_ins: dimension of the input to the DBN
4545
46-
:type n_layers_sizes: list of ints
47-
:param n_layers_sizes: intermediate layers size, must contain
46+
:type hidden_layers_sizes: list of ints
47+
:param hidden_layers_sizes: intermediate layers size, must contain
4848
at least one value
4949
5050
:type n_outs: int
@@ -263,8 +263,8 @@ def test_DBN(finetune_lr=0.1, pretraining_epochs=100,
263263
264264
This is demonstrated on MNIST.
265265
266-
:type learning_rate: float
267-
:param learning_rate: learning rate used in the finetune stage
266+
:type finetune_lr: float
267+
:param finetune_lr: learning rate used in the finetune stage
268268
:type pretraining_epochs: int
269269
:param pretraining_epochs: number of epoch to do pretraining
270270
:type pretrain_lr: float

0 commit comments

Comments
 (0)