vaeac_get_optimizer {shapr} | R Documentation |
Function to create the optimizer used to train vaeac
Description
Only torch::optim_adam()
is currently supported. But it is easy to add an additional option later.
Usage
vaeac_get_optimizer(vaeac_model, lr, optimizer_name = "adam")
Arguments
vaeac_model |
A |
lr |
Positive numeric (default is |
optimizer_name |
String containing the name of the |
Value
A torch::optim_adam()
optimizer connected to the parameters of the vaeac_model
.
Author(s)
Lars Henry Berge Olsen
[Package shapr version 1.0.4 Index]