Optimizer.adam(model, dataloader, loss, ?lr, ?beta1, ?beta2, ?eps, ?weightDecay, ?reversible, ?iters, ?epochs, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix, ?printNewLine)
|
Parameter Types:
-
model
:
Model
-
dataloader
:
DataLoader
-
loss
:
Tensor -> Tensor -> Tensor
-
?lr
:
Tensor
-
?beta1
:
Tensor
-
?beta2
:
Tensor
-
?eps
:
Tensor
-
?weightDecay
:
Tensor
-
?reversible
:
bool
-
?iters
:
int
-
?epochs
:
int
-
?threshold
:
double
-
?print
:
bool
-
?printEvery
:
int
-
?printPrefix
:
string
-
?printPostfix
:
string
-
?printNewLine
:
bool
|
Optimizer.adam(f, x0, ?lr, ?beta1, ?beta2, ?eps, ?iters, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix, ?printNewLine)
|
Parameter Types:
-
f
:
Tensor -> Tensor
-
x0
:
Tensor
-
?lr
:
Tensor
-
?beta1
:
Tensor
-
?beta2
:
Tensor
-
?eps
:
Tensor
-
?iters
:
int
-
?threshold
:
double
-
?print
:
bool
-
?printEvery
:
int
-
?printPrefix
:
string
-
?printPostfix
:
string
-
?printNewLine
:
bool
Return Type:
Tensor * Tensor
|
Optimizer.sgd(model, dataloader, loss, ?lr, ?momentum, ?nesterov, ?weightDecay, ?reversible, ?iters, ?epochs, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix, ?printNewLine)
|
Parameter Types:
-
model
:
Model
-
dataloader
:
DataLoader
-
loss
:
Tensor -> Tensor -> Tensor
-
?lr
:
Tensor
-
?momentum
:
Tensor
-
?nesterov
:
bool
-
?weightDecay
:
Tensor
-
?reversible
:
bool
-
?iters
:
int
-
?epochs
:
int
-
?threshold
:
double
-
?print
:
bool
-
?printEvery
:
int
-
?printPrefix
:
string
-
?printPostfix
:
string
-
?printNewLine
:
bool
|
Optimizer.sgd(f, x0, ?lr, ?momentum, ?nesterov, ?iters, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix, ?printNewLine)
|
Parameter Types:
-
f
:
Tensor -> Tensor
-
x0
:
Tensor
-
?lr
:
Tensor
-
?momentum
:
Tensor
-
?nesterov
:
bool
-
?iters
:
int
-
?threshold
:
double
-
?print
:
bool
-
?printEvery
:
int
-
?printPrefix
:
string
-
?printPostfix
:
string
-
?printNewLine
:
bool
Return Type:
Tensor * Tensor
|