load_state_dict | R Documentation |
This function should only be used to load models saved in python.
For it to work correctly you need to use torch.save
with the flag:
_use_new_zipfile_serialization=True
and also remove all nn.Parameter
classes from the tensors in the dict.
load_state_dict(path, ..., legacy_stream = FALSE)
path |
to the state dict file |
... |
additional arguments that are currently not used. |
legacy_stream |
if |
The above might change with development of this in pytorch's C++ api.
a named list of tensors.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.