mx.model.buckets
¶
Description¶
Train RNN with bucket support
Usage¶
mx.model.buckets(
symbol,
train.data,
eval.data = NULL,
metric = NULL,
arg.params = NULL,
aux.params = NULL,
fixed.params = NULL,
num.round = 1,
begin.round = 1,
initializer = mx.init.uniform(0.01),
optimizer = "sgd",
ctx = NULL,
batch.end.callback = NULL,
epoch.end.callback = NULL,
kvstore = "local",
verbose = TRUE,
metric_cpu = TRUE
)
Arguments¶
Argument |
Description |
---|---|
|
Symbol or list of Symbols representing the model |
|
Training data created by mx.io.bucket.iter |
|
Evaluation data created by mx.io.bucket.iter |
|
int, number of epoch |
|