vaeac_get_current_save_state {shapr} | R Documentation |
Function that extracts additional objects from the environment to the state list
Description
The function extract the objects that we are going to save together with the vaeac
model to make it possible to
train the model further and to evaluate it.
The environment should be the local environment inside the vaeac_train_model_auxiliary()
function.
Usage
vaeac_get_current_save_state(environment)
Arguments
environment |
The |
Value
List containing the values of epoch
, train_vlb
, val_iwae
, val_iwae_running
,
and the state_dict()
of the vaeac model and optimizer.
Author(s)
Lars Henry Berge Olsen
[Package shapr version 1.0.4 Index]