callback_save_model_state_dict: callback_save_model_state_dict

callback_save_model_state_dictR Documentation

callback_save_model_state_dict

Description

Save the model state dictionary to disk at the end of each epoch as a .pt file. For use within luz training loop.

Usage

callback_save_model_state_dict(save_dir = ".", prefix = "model_epoch")

Arguments

save_dir

path and directory to save state dictionary files

prefix

prefix for file name. Default is "model_epoch". The suffix is the epoch number.

Value

No R object is returned. The state dictionary is saved to disk as a .pt file.


geodl documentation built on Nov. 12, 2025, 5:07 p.m.