Default pretrained weights to eval mode (#1843)

cd8f87bd · moto · GitHub · 78c382ee · cd8f87bd
Unverified Commit cd8f87bd authored Oct 08, 2021 by moto Committed by GitHub Oct 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

torchaudio/models/wav2vec2/pretrained.py torchaudio/models/wav2vec2/pretrained.py +1 -0

No files found.
--- a/torchaudio/models/wav2vec2/pretrained.py
+++ b/torchaudio/models/wav2vec2/pretrained.py
@@ -72,6 +72,7 @@ class Wav2Vec2PretrainedModelBundle:
        dl_kwargs = {} if dl_kwargs is None else dl_kwargs
        state_dict = load_state_dict_from_url(url, **dl_kwargs)
        model.load_state_dict(state_dict)
+        model.eval()
        return model

    def get_labels(