TBD
Static member | Description |
Full Usage:
optim.adam (model, dataloader, loss, ?lr, ?beta1, ?beta2, ?eps, ?weightDecay, ?reversible, ?iters, ?epochs, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix)
Parameters:
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
|
|
Full Usage:
optim.adam (f, x0, ?lr, ?beta1, ?beta2, ?eps, ?iters, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix)
Parameters:
Tensor -> Tensor
x0 : Tensor
?lr : Tensor
?beta1 : Tensor
?beta2 : Tensor
?eps : Tensor
?iters : int
?threshold : double
?print : bool
?printEvery : int
?printPrefix : string
?printPostfix : string
Returns: Tensor * Tensor
|
|
Full Usage:
optim.sgd (model, dataloader, loss, ?lr, ?momentum, ?nesterov, ?weightDecay, ?reversible, ?iters, ?epochs, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix)
Parameters:
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
|
|
Full Usage:
optim.sgd (f, x0, ?lr, ?momentum, ?nesterov, ?iters, ?threshold, ?print, ?printEvery, ?printPrefix, ?printPostfix)
Parameters:
Tensor -> Tensor
x0 : Tensor
?lr : Tensor
?momentum : Tensor
?nesterov : bool
?iters : int
?threshold : double
?print : bool
?printEvery : int
?printPrefix : string
?printPostfix : string
Returns: Tensor * Tensor
|
© Copyright 2021, DiffSharp Contributors.