vaeac_get_current_save_state function

Function that extracts additional objects from the environment to the state list

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.

vaeac_get_current_save_state(environment)

Arguments

• environment: The base::environment() where the objects are stored.

Returns

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