Skip to content

GitLab

Commit cb77a86c authored by moto's avatar moto
Browse files

Default pretrained weights to eval mode (#1843)

parent 486022e9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
torchaudio/models/wav2vec2/pretrained.py
View file @ cb77a86c
...@@ -72,6 +72,7 @@ class Wav2Vec2PretrainedModelBundle: ...@@ -72,6 +72,7 @@ class Wav2Vec2PretrainedModelBundle:
dl_kwargs = {} if dl_kwargs is None else dl_kwargs dl_kwargs = {} if dl_kwargs is None else dl_kwargs
state_dict = load_state_dict_from_url(url, **dl_kwargs) state_dict = load_state_dict_from_url(url, **dl_kwargs)
model.load_state_dict(state_dict) model.load_state_dict(state_dict)
model.eval()
return model return model
def get_labels( def get_labels(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment