mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Use eval() for the masked LM model in decoding.
This commit is contained in:
parent
7f5d9a1671
commit
b8037cd529
@ -754,6 +754,7 @@ def main():
|
|||||||
)
|
)
|
||||||
masked_lm_model.to(device2)
|
masked_lm_model.to(device2)
|
||||||
masked_lm_model.device = device2
|
masked_lm_model.device = device2
|
||||||
|
masked_lm_model.eval()
|
||||||
else:
|
else:
|
||||||
masked_lm_model = None
|
masked_lm_model = None
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user