vaeac_get_current_save_state: Function that extracts additional objects from the...

View source: R/approach_vaeac.R

vaeac_get_current_save_stateR 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 base::environment() where the objects are stored.

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


NorskRegnesentral/shapr documentation built on April 19, 2024, 1:19 p.m.