argument
|
meaning
|
default
value / comment |
| --train-data |
training
data
specification |
required |
| --nnet-spec |
--nnet-spec="d:h(1):h(2):...:h(n):s"
Eg.250:1024:1024:1024:1024:1920
|
required. d-input
dimension; h(i)-size of the
i-th hidden layers; s-number
of targets |
| --wdir |
working
directory |
required |
|
| --param-output-file |
path
to
save model parameters in the PDNN format |
by
default
"": doesn't output PDNN-formatted model |
| --cfg-output-file |
path
to
save model config |
by
default
"": doesn't output model config |
| --kaldi-output-file |
path
to
save the Kaldi-formatted model |
by
default
"": doesn't output Kaldi-formatted model |
|
| --learning-rate |
learning
rate for Bernoulli-Bernoulli RBM
|
by default 0.08 |
| --gbrbm-learning-rate |
learning
rate for Gaussian-Bernoulli RBM
|
by default 0.005 |
| --epoch-number |
number
of
epochs |
by
default
10
|
| --batch-size |
mini-batch
size during training |
by default 128 |
| --momentum |
momentum
string with the format "init:final:init_epochs"
E.g.: "0.5:0.9:5"
|
by
default 0.5:0.9:5, it has 3 fields:
init -- initial momentum
final -- final momentum
init_epochs -- for how many epochs we use initial_momentum
|
| --ptr-layer-number |
number
of layers to be trained
|
by
default train all
the hidden layers |
|
| --first-layer-type |
type
for the first layer; either "bb" (Bernoulli-Bernoulli) or
"gb" (Gaussian-Bernoulli)
|
by
default "gb"
|