mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-10 22:45:27 +00:00
Fix an error introduced while supporting torchscript. (#134)
Should be `G.dummy = 1`, not `G["dummy"] = 1`.
This commit is contained in:
parent
0e541f5b5d
commit
243fb9723c
@ -615,7 +615,7 @@ def main():
|
|||||||
# Save a dummy value so that it can be loaded in C++.
|
# Save a dummy value so that it can be loaded in C++.
|
||||||
# See https://github.com/pytorch/pytorch/issues/67902
|
# See https://github.com/pytorch/pytorch/issues/67902
|
||||||
# for why we need to do this.
|
# for why we need to do this.
|
||||||
G["dummy"] = 1
|
G.dummy = 1
|
||||||
|
|
||||||
torch.save(G.as_dict(), params.lm_dir / "G_4_gram.pt")
|
torch.save(G.as_dict(), params.lm_dir / "G_4_gram.pt")
|
||||||
else:
|
else:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user