sjSDMControl {sjSDM}

sjSDM control object

Description

sjSDM control object

Usage

sjSDMControl(
  optimizer = RMSprop(),
  scheduler = 0,
  lr_reduce_factor = 0.99,
  early_stopping_training = 0,
  mixed = FALSE
)

Arguments

optimizer

object of type RMSprop, Adamax, SGD, AccSGD, madgrad, or AdaBound

scheduler

reduce lr on plateau scheduler or not (0 means no scheduler, > 0 number of epochs before reducing learning rate)

lr_reduce_factor

factor to reduce learning rate in scheduler

early_stopping_training

number of epochs without decrease in training loss before invoking early stopping (0 means no early stopping).

mixed

mixed (half-precision) training or not. Only recommended for GPUs > 2000 series

Value

List with the following fields:

optimizer

Function which returns an optimizer.

scheduler_boolean

Logical, use scheduler or not.

scheduler_patience

Integer, number of epochs to wait before applying plateau scheduler.

lr_reduce_factor

Numerical, learning rate reduce factor.

mixed

Logical, use mixed training or not.

early_stopping_training

Numerical, early stopping after n epochs.