mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Fix an error introduced while supporting torchscript.
Should be `G.dummy = 1`, not `G["dummy"] = 1`.
This commit is contained in:
parent
769a9791ec
commit
277654249d
@ -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