| callback_save_model_state_dict | R Documentation |
Save the model state dictionary to disk at the end of each epoch as a .pt file. For use within luz training loop.
callback_save_model_state_dict(save_dir = ".", prefix = "model_epoch")
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. |
No R object is returned. The state dictionary is saved to disk as a .pt file.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.