vaeac
model and set it in the right state and modeFunction to load a vaeac
model and set it in the right state and mode
vaeac_get_model_from_checkp(checkpoint, cuda, mode_train)
A vaeac
model with the correct state (based on checkpoint
), sent to the desired hardware (based on
cuda
), and in the right mode (based on mode_train
).
List. This must be a loaded vaeac
save object. That is, torch::torch_load('vaeac_save_path')
.
Logical (default is FALSE
). If TRUE
, then the vaeac
model will be trained using cuda/GPU.
If torch::cuda_is_available()
is FALSE
, the we fall back to use CPU. If FALSE
, we use the CPU. Using a GPU
for smaller tabular dataset often do not improve the efficiency.
See vignette("installation", package = "torch")
fo help to enable running on the GPU (only Linux and Windows).
Logical. If TRUE
, the returned vaeac
model is set to be in training mode.
If FALSE
, the returned vaeac
model is set to be in evaluation mode.
Lars Henry Berge Olsen