load_state_dict: Load a state dict file
In torch: Tensors and Neural Networks with 'GPU' Acceleration
load_state_dict R Documentation
Load a state dict file
Description
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.
Usage
load_state_dict(path, ..., legacy_stream = FALSE)
Arguments
path
to the state dict file
...
additional arguments that are currently not used.
legacy_stream
if TRUE then the state dict is loaded using a
a legacy way of handling streams.
Details
The above might change with development of this
in pytorch's C++ api.
Value
a named list of tensors.
torch documentation built on May 29, 2024, 9:54 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
| load_state_dict | R Documentation |
Load a state dict file
Description
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.
Usage
load_state_dict(path, ..., legacy_stream = FALSE)
Arguments
path |
to the state dict file |
... |
additional arguments that are currently not used. |
legacy_stream |
if |
Details
The above might change with development of this in pytorch's C++ api.
Value
a named list of tensors.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.