Zero'ing params docs
This commit is contained in:
+6
-5
@@ -2,14 +2,15 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
"""Saves the model as a TorchScript.
|
"""Saves the model as a TorchScript.
|
||||||
|
|
||||||
The resulting file can be loaded in C++ code and then used for training or inference with:
|
The resulting file can be loaded in C++ code and then used for training or
|
||||||
|
inference with:
|
||||||
#include <torch/script.h>
|
#include <torch/script.h>
|
||||||
torch::jit::Module module = torch::jit::load("model.pt")
|
torch::jit::Module module = torch::jit::load("model.pt")
|
||||||
|
|
||||||
Note that the model includes the initial parameters and with default ModelArgs the serialized model
|
Note that the model includes the initial parameters and with default ModelArgs the
|
||||||
is 59M and gzips down to 55M. If you want to serialize/distribute the model parameters separately
|
serialized model is 59M and gzips down to 55M. If you want to serialize/distribute the
|
||||||
and the size of the model file you can zero out the parameters before saving it and it will gzip
|
model parameters separately and the size of the model file you can zero out the
|
||||||
down to 780K:
|
parameters before saving it and it will gzip down to 780K:
|
||||||
for p in model.parameters():
|
for p in model.parameters():
|
||||||
p.detach().zero_()
|
p.detach().zero_()
|
||||||
"""
|
"""
|
||||||
|
|||||||
Reference in New Issue
Block a user