rpudlPretrain {rpud}R Documentation

RPUDL Model Pre-training

Description

Pre-train a RPUDL deep learning model.

Usage

rpudlPretrain(
    model,
    learning.rate = 0,
    batch.size = 128,
    iterations = 128,
    display = 100,
    seed = 0,
    ...,
    log.level = 1
    )

Arguments

model

a rpudl model object to be pre-trained

learning.rate

manual override of the learning rate, ignored if zero or negative

batch.size

the size of a data batch for each training iteration

iterations

maximum number of training iterations

display

a logical to display training progress

seed

seed of random number generation

...

further arguments to be passed to or from methods

log.level

level of the method output (0 = silent, 1 = basic output, 2 = detailed output)

Details

Only the fully-connected layers will be pre-trained via one of the following mechanisms:

Value

An rpudl object with pre-trained parameters.

Author(s)

Chi Yau
chi.yau@r-tutor.com

See Also

rpudl,
rpudlTrain,
rpudlPredict,

Examples

## Not run: 
# create data source
ds <- rpudlCreateDataSource(
        data.format="lmdb",
        data.dir="data/mnist",
        train.data="mnist-official-data_train_lmdb",
        test.data="mnist-official-data_test_lmdb",
        data.shape=c(28, 28)
)

# create model
model <- rpudl(
        "masked_autoencoder.prototxt",
        data.source=ds
)   

# pre-train
model <- rpudlPretrain(model, batch=100, iter=1000)

## End(Not run)

[Package rpud version 0.7.0 Index]